yolov8-detection-lego-demo

Sleeping

CWrecker commited on Apr 27, 2023

Commit

c8d1bd7

•

1 Parent(s): 2a392dd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,6 +26,7 @@ def inference(path:str, threshold:float=0.6):
                 color=(0, 0, 255),
                 thickness=2,
             )
             counter+=1
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB), counter

                 color=(0, 0, 255),
                 thickness=2,
             )
+            cv2.putText(image, cls, (int(box[0]), int(box[2])))
             counter+=1
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB), counter