sabrinabenas commited on
Commit
0d79970
β€’
1 Parent(s): 9a8a433
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -160,10 +160,11 @@ def predict_dlc(list_np_crops,
160
  # scale crop here?
161
  keypts_xyp = dlc_live.get_pose(crop) # third column is llk!
162
  # set kpts below threhsold to nan
 
163
  keypts_xyp[keypts_xyp[:,-1] < kpts_likelihood_th,:] = np_aux.fill(np.nan)
164
  # add kpts of this crop to list
165
  list_kpts_per_crop.append(keypts_xyp)
166
-
167
  return list_kpts_per_crop
168
 
169
 
 
160
  # scale crop here?
161
  keypts_xyp = dlc_live.get_pose(crop) # third column is llk!
162
  # set kpts below threhsold to nan
163
+ print(keypts_xyp)
164
  keypts_xyp[keypts_xyp[:,-1] < kpts_likelihood_th,:] = np_aux.fill(np.nan)
165
  # add kpts of this crop to list
166
  list_kpts_per_crop.append(keypts_xyp)
167
+ #return confidence here
168
  return list_kpts_per_crop
169
 
170