Spaces:
Build error
Build error
sabrinabenas
commited on
Commit
•
8a9e330
1
Parent(s):
2f5c6bc
test
Browse files
app.py
CHANGED
|
@@ -113,7 +113,7 @@ def crop_animal_detections(yolo_results,
|
|
| 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()
|
|
|
|
| 113 |
|
| 114 |
## Extract animal crops
|
| 115 |
#print(yolo_results)
|
| 116 |
+
list_labels_as_str = [i for i in yolo_results.names.values()] # ['animal', 'person', 'vehicle']
|
| 117 |
list_np_animal_crops = []
|
| 118 |
# for every image
|
| 119 |
#pdb.set_trace()
|