Reacher commited on
Commit
8527527
·
1 Parent(s): 7a873d2
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -12,10 +12,14 @@ def show_preds_image(image_path):
12
  print(image_path)
13
  image = cv2.imread(image_path)
14
  outputs = model.predict(source=image_path)
15
- #print(outputs[0])
 
 
16
  results = outputs[0].cpu().numpy()
17
  #for i, det in enumerate(results.boxes.xyxy):
18
  #cls = TargetMapper[results.boxes.cls[0]]
 
 
19
  det = results.boxes.xyxy[0]
20
  #print(det)
21
  #print(cls)
 
12
  print(image_path)
13
  image = cv2.imread(image_path)
14
  outputs = model.predict(source=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)
22
+ print(results[0].boxes)
23
  det = results.boxes.xyxy[0]
24
  #print(det)
25
  #print(cls)