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