Spaces:
Build error
Build error
sabrinabenas
commited on
Commit
β’
8aa3d1c
1
Parent(s):
8a9e330
test
Browse files
app.py
CHANGED
@@ -134,7 +134,7 @@ def crop_animal_detections(yolo_results,
|
|
134 |
pred_label = det_array[j,5]
|
135 |
|
136 |
# keep animal crops above threshold
|
137 |
-
pdb.set_trace()
|
138 |
if (pred_label == list_labels_as_str.index('animal')) and \
|
139 |
(pred_llk >= likelihood_th):
|
140 |
area = (xmin_rd, ymin_rd, xmax_rd, ymax_rd)
|
@@ -164,7 +164,7 @@ def predict_dlc(list_np_crops,
|
|
164 |
keypts_xyp = dlc_live.get_pose(crop) # third column is llk!
|
165 |
# set kpts below threhsold to nan
|
166 |
print(keypts_xyp)
|
167 |
-
pdb.set_trace()
|
168 |
keypts_xyp[keypts_xyp[:,-1] < kpts_likelihood_th,:] = np_aux.fill(np.nan)
|
169 |
# add kpts of this crop to list
|
170 |
list_kpts_per_crop.append(keypts_xyp)
|
@@ -205,7 +205,7 @@ def predict_pipeline(img_input,
|
|
205 |
md_results = predict_md(img_input,
|
206 |
mega_model_input,
|
207 |
size=640) #Image.fromarray(results.imgs[0])
|
208 |
-
|
209 |
################################################################
|
210 |
# Obtain animal crops for bboxes with confidence above th
|
211 |
|
|
|
134 |
pred_label = det_array[j,5]
|
135 |
|
136 |
# keep animal crops above threshold
|
137 |
+
#pdb.set_trace()
|
138 |
if (pred_label == list_labels_as_str.index('animal')) and \
|
139 |
(pred_llk >= likelihood_th):
|
140 |
area = (xmin_rd, ymin_rd, xmax_rd, ymax_rd)
|
|
|
164 |
keypts_xyp = dlc_live.get_pose(crop) # third column is llk!
|
165 |
# set kpts below threhsold to nan
|
166 |
print(keypts_xyp)
|
167 |
+
#pdb.set_trace()
|
168 |
keypts_xyp[keypts_xyp[:,-1] < kpts_likelihood_th,:] = np_aux.fill(np.nan)
|
169 |
# add kpts of this crop to list
|
170 |
list_kpts_per_crop.append(keypts_xyp)
|
|
|
205 |
md_results = predict_md(img_input,
|
206 |
mega_model_input,
|
207 |
size=640) #Image.fromarray(results.imgs[0])
|
208 |
+
pdb.set_trace()
|
209 |
################################################################
|
210 |
# Obtain animal crops for bboxes with confidence above th
|
211 |
|