Spaces:
Runtime error
Runtime error
fix the bug finally
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ def show_preds_image(image_path):
|
|
13 |
results = outputs[0]
|
14 |
print("results:", results)
|
15 |
|
16 |
-
for i, det in enumerate(results.boxes.xyxy):
|
17 |
cv2.rectangle(
|
18 |
image,
|
19 |
(int(det[0]), int(det[1])),
|
@@ -46,8 +46,8 @@ def show_preds_video(video_path):
|
|
46 |
if ret:
|
47 |
frame_copy = frame.copy()
|
48 |
outputs = model.predict(source=frame)
|
49 |
-
results = outputs[0]
|
50 |
-
for i, det in enumerate(results.boxes.xyxy):
|
51 |
cv2.rectangle(
|
52 |
frame_copy,
|
53 |
(int(det[0]), int(det[1])),
|
|
|
13 |
results = outputs[0]
|
14 |
print("results:", results)
|
15 |
|
16 |
+
for i, det in enumerate(results.boxes.xyxy.cpu().numpy()):
|
17 |
cv2.rectangle(
|
18 |
image,
|
19 |
(int(det[0]), int(det[1])),
|
|
|
46 |
if ret:
|
47 |
frame_copy = frame.copy()
|
48 |
outputs = model.predict(source=frame)
|
49 |
+
results = outputs[0]
|
50 |
+
for i, det in enumerate(results.boxes.xyxy.cpu().numpy()):
|
51 |
cv2.rectangle(
|
52 |
frame_copy,
|
53 |
(int(det[0]), int(det[1])),
|