Spaces:

hashb
/

object-detection-yolo

Build error

hashb commited on Sep 9, 2022

Commit

c794e33

1 Parent(s): 9941aa0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ model.setInputParams(scale=1 / 255, size=(416, 416), swapRB=True)
 def detect(image):
     img = Image.fromarray(image)
-    img = cv2.cvtColor(np.array(img), COLOR_RGBA2BGRA)
     classIds, scores, boxes = model.detect(img, confThreshold=0.6, nmsThreshold=0.4)
     for (classId, score, box) in zip(classIds, scores, boxes):

 def detect(image):
     img = Image.fromarray(image)
+    img = cv2.cvtColor(np.array(img), COLOR_RGB2BGR)
     classIds, scores, boxes = model.detect(img, confThreshold=0.6, nmsThreshold=0.4)
     for (classId, score, box) in zip(classIds, scores, boxes):