mwmathis commited on
Commit
d1d45e2
Β·
1 Parent(s): bb68d34

Update dlc_utils.py

Browse files
Files changed (1) hide show
  1. dlc_utils.py +1 -1
dlc_utils.py CHANGED
@@ -12,7 +12,7 @@ def predict_dlc(list_np_crops,
12
  dlc_proc):
13
 
14
  # run dlc thru list of crops
15
- dlc_live = DLCLive(DLCmodel[0], processor=dlc_proc)
16
  dlc_live.init_inference(list_np_crops[0])
17
 
18
  list_kpts_per_crop = []
 
12
  dlc_proc):
13
 
14
  # run dlc thru list of crops
15
+ dlc_live = DLCLive("superanimal_quadruped", processor=dlc_proc)
16
  dlc_live.init_inference(list_np_crops[0])
17
 
18
  list_kpts_per_crop = []