yolopv2

Sleeping

hank1996 commited on Sep 1, 2022

Commit

d86e92c

1 Parent(s): fc84157

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -147,7 +147,7 @@ def detect(img,model):
             nms_time.update(t4-t3,img.size(0))
             det=det_pred[0]
-            save_path = str(save_dir +'/'+ Path(path).name)
             _, _, height, width = img.shape
             h,w,_=img_det.shape

             nms_time.update(t4-t3,img.size(0))
             det=det_pred[0]
+            save_path = str(save_dir +'/'+ 'img.jpg')
             _, _, height, width = img.shape
             h,w,_=img_det.shape