ysr commited on
Commit
a2db237
·
1 Parent(s): 36908fb

fix the bug finally

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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].cpu().numpy()
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])),