Reacher commited on
Commit
0a375d6
·
1 Parent(s): 8527527
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def show_preds_image(image_path):
15
  print(outputs)
16
  print(outputs[0])
17
 
18
- results = outputs[0].cpu().numpy()
19
  #for i, det in enumerate(results.boxes.xyxy):
20
  #cls = TargetMapper[results.boxes.cls[0]]
21
  print(results.boxes)
 
15
  print(outputs)
16
  print(outputs[0])
17
 
18
+ results = outputs[0]#.cpu().numpy()
19
  #for i, det in enumerate(results.boxes.xyxy):
20
  #cls = TargetMapper[results.boxes.cls[0]]
21
  print(results.boxes)