зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-11-04 16:06:05 +02:00
b585097f19
* reduce code smells * account for transl. version change * remove print statements * remove imageai
42 строки
1.1 KiB
Python
42 строки
1.1 KiB
Python
from misinformation.utils import AnalysisMethod
|
|
from misinformation.objects_cvlib import ObjectCVLib
|
|
from misinformation.objects_cvlib import init_default_objects
|
|
|
|
|
|
class ObjectDetectorClient(AnalysisMethod):
|
|
def __init__(self):
|
|
# The detector is default to CVLib
|
|
# Here other libraries can be added
|
|
self.detector = ObjectCVLib()
|
|
|
|
def set_client_to_cvlib(self):
|
|
self.detector = ObjectCVLib()
|
|
|
|
def analyse_image(self, subdict=None):
|
|
"""Localize objects in the local image.
|
|
|
|
Args:
|
|
subdict: The dictionary for an image expression instance.
|
|
"""
|
|
|
|
return self.detector.analyse_image(subdict)
|
|
|
|
|
|
class ObjectDetector(AnalysisMethod):
|
|
od_client = ObjectDetectorClient()
|
|
|
|
def __init__(self, subdict: dict):
|
|
super().__init__(subdict)
|
|
self.subdict.update(self.set_keys())
|
|
|
|
def set_keys(self):
|
|
return init_default_objects()
|
|
|
|
def analyse_image(self):
|
|
self.subdict = ObjectDetector.od_client.analyse_image(self.subdict)
|
|
return self.subdict
|
|
|
|
@staticmethod
|
|
def set_client_to_cvlib():
|
|
ObjectDetector.od_client.set_client_to_cvlib()
|