sabrinabenas commited on
Commit
a43df40
β€’
1 Parent(s): 64e43c8
Files changed (1) hide show
  1. app.py +2 -0
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)