ee48f141f8
Merge in RR/image-prediction from RED-5107-robustify-image-service-alternative to release/1.2.x
Squashed commit of the following:
commit 1a8fbeebd3c05f25d69210e53bf6dce67bc2342f
Merge: 00ac0d6 c03913e
Author: Matthias Bisping <matthias.bisping@axbit.com>
Date: Tue Aug 30 16:19:16 2022 +0200
Merge branch 'release/1.2.x' into RED-5107-robustify-image-service-alternative
commit 00ac0d61abdd97eb7c2576d2db9e6859b91c9c41
Author: Matthias Bisping <matthias.bisping@axbit.com>
Date: Tue Aug 30 16:03:41 2022 +0200
applied black
commit 983265f4355253a3a371747b04b1926ff3578fef
Author: Matthias Bisping <matthias.bisping@axbit.com>
Date: Tue Aug 30 15:59:11 2022 +0200
Added image validation after image extraction to parsable-pdf image extractor. Invalid images are dropped, hence these images will appear as skipped for the service caller.
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import abc
|
|
|
|
import numpy as np
|
|
import tensorflow as tf
|
|
|
|
|
|
class ModelWrapper(abc.ABC):
|
|
def __init__(self, classes, base_weights_path=None, weights_path=None):
|
|
self.__classes = classes
|
|
self.model = self.__build(base_weights_path)
|
|
self.model.load_weights(weights_path)
|
|
|
|
@property
|
|
@abc.abstractmethod
|
|
def input_shape(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
def classes(self):
|
|
return self.__classes
|
|
|
|
@abc.abstractmethod
|
|
def __preprocess_tensor(self, tensor):
|
|
raise NotImplementedError
|
|
|
|
@staticmethod
|
|
def __images_to_tensor(images):
|
|
return np.array(list(map(tf.keras.preprocessing.image.img_to_array, images)))
|
|
|
|
def __resize_and_convert(self, image):
|
|
return image.resize(self.input_shape[:-1]).convert("RGB")
|
|
|
|
def prep_images(self, images):
|
|
images = map(self.__resize_and_convert, images)
|
|
tensor = self.__images_to_tensor(images)
|
|
tensor = self.__preprocess_tensor(tensor)
|
|
|
|
return tensor
|
|
|
|
@abc.abstractmethod
|
|
def __build(self, base_weights=None) -> tf.keras.models.Model:
|
|
raise NotImplementedError
|