faultdetection3 / services /detection_service.py
Sanjayraju30's picture
Update services/detection_service.py
8f3c13e verified
raw
history blame contribute delete
283 Bytes
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]