Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

applied black

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-03 04:47:49 +02:00
parent 8c916a79c3
commit ab382646b7
7 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
image_prediction/redai_adapter/efficient_net_wrapper.py
View File

@ -4,7 +4,6 @@ from image_prediction.redai_adapter.model_wrapper import ModelWrapper
class EfficientNetWrapper(ModelWrapper): class EfficientNetWrapper(ModelWrapper):
def __init__(self, classes, base_weights_path=None, weights_path=None): def __init__(self, classes, base_weights_path=None, weights_path=None):
self.__input_shape = (224, 224, 3) self.__input_shape = (224, 224, 3)
super().__init__(classes=classes, base_weights_path=base_weights_path, weights_path=weights_path) super().__init__(classes=classes, base_weights_path=base_weights_path, weights_path=weights_path)

1
image_prediction/redai_adapter/model_wrapper.py
View File

@ -5,7 +5,6 @@ import tensorflow as tf
class ModelWrapper(abc.ABC): class ModelWrapper(abc.ABC):
def __init__(self, classes, base_weights_path=None, weights_path=None): def __init__(self, classes, base_weights_path=None, weights_path=None):
self.__classes = classes self.__classes = classes
self.model = self.__build(base_weights_path) self.model = self.__build(base_weights_path)

1
image_prediction/transformer/transformers/coordinate/coordinate_transformer.py
View File

@ -4,7 +4,6 @@ from image_prediction.transformer.transformer import Transformer
class CoordinateTransformer(Transformer): class CoordinateTransformer(Transformer):
@abc.abstractmethod @abc.abstractmethod
def _forward(self, metadata): def _forward(self, metadata):
raise NotImplementedError raise NotImplementedError

1
image_prediction/transformer/transformers/coordinate/fitz.py
View File

@ -2,7 +2,6 @@ from image_prediction.transformer.transformers.coordinate.coordinate_transformer
class FitzCoordinateTransformer(CoordinateTransformer): class FitzCoordinateTransformer(CoordinateTransformer):
def _forward(self, metadata: dict): def _forward(self, metadata: dict):
"""Fitz uses top left corner as origin; we take this as the reference coordinate system.""" """Fitz uses top left corner as origin; we take this as the reference coordinate system."""
return metadata return metadata

1
image_prediction/transformer/transformers/coordinate/fpdf.py
View File

@ -2,7 +2,6 @@ from image_prediction.transformer.transformers.coordinate.coordinate_transformer
class FPDFCoordinateTransformer(CoordinateTransformer): class FPDFCoordinateTransformer(CoordinateTransformer):
def _forward(self, metadata: dict): def _forward(self, metadata: dict):
"""FPDF uses top left corner as origin; we take this as the reference coordinate system.""" """FPDF uses top left corner as origin; we take this as the reference coordinate system."""
return metadata return metadata

1
image_prediction/transformer/transformers/coordinate/pdfnet.py
View File

@ -7,7 +7,6 @@ from image_prediction.transformer.transformers.coordinate.coordinate_transformer
class PDFNetCoordinateTransformer(CoordinateTransformer): class PDFNetCoordinateTransformer(CoordinateTransformer):
def _forward(self, metadata: dict): def _forward(self, metadata: dict):
"""PDFNet coordinate system origin is in the bottom left corner.""" """PDFNet coordinate system origin is in the bottom left corner."""
y1, y2, page_height = itemgetter(Info.Y1, Info.Y2, Info.PAGE_HEIGHT)(metadata) y1, y2, page_height = itemgetter(Info.Y1, Info.Y2, Info.PAGE_HEIGHT)(metadata)

4
image_prediction/utils/pdf_annotation.py
View File

@ -54,10 +54,10 @@ def annotate_image(doc, image_info):
add_note() add_note()
def init(): def init():
PDFNet.Initialize( PDFNet.Initialize(
"Knecon AG(en.knecon.swiss):OEM:DDA-R::WL+:AMS(20211029):BECC974307DAB4F34B513BC9B2531B24496F6FCB83CD8AC574358A959730B622FABEF5C7") "Knecon AG(en.knecon.swiss):OEM:DDA-R::WL+:AMS(20211029):BECC974307DAB4F34B513BC9B2531B24496F6FCB83CD8AC574358A959730B622FABEF5C7"
)
def draw_metadata_box(pdf_path, metadata, store_path): def draw_metadata_box(pdf_path, metadata, store_path):