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

...

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-02 02:45:21 +02:00
parent 5c23898280
commit 613bba8cfc
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
image_prediction/classifier/classifier.py
View File

@ -30,5 +30,5 @@ class Classifier:
return list(self.__pipe(batch)) return list(self.__pipe(batch))
def __call__(self, batch: np.array) -> List[str]: def __call__(self, batch: np.array) -> List[str]:
logger.debug("Classifier.predicting") logger.debug("Classifier.predict")
return self.predict(batch) return self.predict(batch)

3
image_prediction/formatter/formatters/camel_case.py
View File

@ -2,8 +2,7 @@ from image_prediction.formatter.formatters.key_formatter import KeyFormatter
class Snake2CamelCaseKeyFormatter(KeyFormatter): class Snake2CamelCaseKeyFormatter(KeyFormatter):
@staticmethod def format_key(self, key):
def format_key(key):
if isinstance(key, str): if isinstance(key, str):
head, *tail = key.split("_") head, *tail = key.split("_")