Create detection_service.py

detection_service.py

@@ -0,0 +1,9 @@
+from transformers import pipeline
+import cv2
+
+object_detector = pipeline("object-detection", model="facebook/detr-resnet-50")
+
+def detect_objects(image_path):
+    image = cv2.imread(image_path)
+    results = object_detector(image)
+    return [r for r in results if r['score'] > 0.7]