Spaces:
Build error
Build error
sabrinabenas
commited on
Commit
β’
5763932
1
Parent(s):
a43df40
change imgs by ims in yolo
Browse files
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.
|
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 |
|