sabrinabenas commited on
Commit
8aa3d1c
β€’
1 Parent(s): 8a9e330
Files changed (1) hide show
  1. app.py +3 -3
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