Spaces:

Gradio-Blocks
/

Gradio_YOLOv5_Det

Runtime error

Zengyf-CVer commited on Jun 3, 2022

Commit

7873a55

1 Parent(s): 53f1c15

v05 update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -419,8 +419,7 @@ def yolo_det_video(video, device, model_name, infer_size, conf, iou, max_num, mo
         if not ret:
             break
-        frame2 = frame.copy()
-        results = model(frame2, size=infer_size)  # detection
         h, w, _ = frame.shape  # frame size
         img_size = (w, h)  # frame size

         if not ret:
             break
+        results = model(frame, size=infer_size)  # detection
         h, w, _ = frame.shape  # frame size
         img_size = (w, h)  # frame size