sabrinabenas commited on
Commit
5763932
β€’
1 Parent(s): a43df40

change imgs by ims in yolo

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -112,12 +112,12 @@ def crop_animal_detections(yolo_results,
112
  likelihood_th):
113
 
114
  ## Extract animal crops
115
- print(yolo_results)
116
  list_labels_as_str = yolo_results.names # ['animal', 'person', 'vehicle']
117
  list_np_animal_crops = []
118
  # for every image
119
- pdb.set_trace()
120
- for img, det_array in zip(yolo_results.imgs,
121
  yolo_results.xyxy):
122
 
123
  # for every detection
@@ -207,6 +207,7 @@ def predict_pipeline(img_input,
207
 
208
  ################################################################
209
  # Obtain animal crops for bboxes with confidence above th
 
210
  list_crops = crop_animal_detections(md_results,
211
  bbox_likelihood_th)
212
 
 
112
  likelihood_th):
113
 
114
  ## Extract animal crops
115
+ #print(yolo_results)
116
  list_labels_as_str = yolo_results.names # ['animal', 'person', 'vehicle']
117
  list_np_animal_crops = []
118
  # for every image
119
+ #pdb.set_trace()
120
+ for img, det_array in zip(yolo_results.ims,
121
  yolo_results.xyxy):
122
 
123
  # for every detection
 
207
 
208
  ################################################################
209
  # Obtain animal crops for bboxes with confidence above th
210
+
211
  list_crops = crop_animal_detections(md_results,
212
  bbox_likelihood_th)
213