Spaces:
Build error
Build error
sabrinabenas
commited on
Commit
β’
a43df40
1
Parent(s):
64e43c8
test0
Browse files
app.py
CHANGED
@@ -116,6 +116,7 @@ def crop_animal_detections(yolo_results,
|
|
116 |
list_labels_as_str = yolo_results.names # ['animal', 'person', 'vehicle']
|
117 |
list_np_animal_crops = []
|
118 |
# for every image
|
|
|
119 |
for img, det_array in zip(yolo_results.imgs,
|
120 |
yolo_results.xyxy):
|
121 |
|
@@ -162,6 +163,7 @@ def predict_dlc(list_np_crops,
|
|
162 |
keypts_xyp = dlc_live.get_pose(crop) # third column is llk!
|
163 |
# set kpts below threhsold to nan
|
164 |
print(keypts_xyp)
|
|
|
165 |
keypts_xyp[keypts_xyp[:,-1] < kpts_likelihood_th,:] = np_aux.fill(np.nan)
|
166 |
# add kpts of this crop to list
|
167 |
list_kpts_per_crop.append(keypts_xyp)
|
|
|
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 |
|
|
|
163 |
keypts_xyp = dlc_live.get_pose(crop) # third column is llk!
|
164 |
# set kpts below threhsold to nan
|
165 |
print(keypts_xyp)
|
166 |
+
pdb.set_trace()
|
167 |
keypts_xyp[keypts_xyp[:,-1] < kpts_likelihood_th,:] = np_aux.fill(np.nan)
|
168 |
# add kpts of this crop to list
|
169 |
list_kpts_per_crop.append(keypts_xyp)
|