async def logo_detection(files: List[UploadFile] = File(...)):
response = None
try:
image_processor = ImageProcessor()
data = json.dumps({image_processor.detect_logos(files[0])})
response = Response(content=data, media_type="application/json")
except Exception as error:
logging.error("Error while uploading file ", str(error))
response = {"Error ": str(error)}
return response
def test_detect_logos(self):
images = [
"altroz.png", "cred.png", "dream11.png", "paytm.png",
"unacademy.png"
]
image_processor = ImageProcessor()
for image in images:
detected_logos = image_processor.detect_logos(image)
detected_logo_names = [logo['name'] for logo in detected_logos]
brand_name = image.split(".")[0]
self.assertIn(brand_name, detected_logo_names)
