jeylau commited on
Commit
a5e6f78
Β·
1 Parent(s): 3bb8aa6

Update dlc_utils.py

Browse files
Files changed (1) hide show
  1. dlc_utils.py +2 -2
dlc_utils.py CHANGED
@@ -8,11 +8,11 @@ from dlclive import DLCLive, Processor
8
  ##########################################
9
  def predict_dlc(list_np_crops,
10
  kpts_likelihood_th,
11
- DLCmodel,
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 = []
 
8
  ##########################################
9
  def predict_dlc(list_np_crops,
10
  kpts_likelihood_th,
11
+ dlc_model_folder,
12
  dlc_proc):
13
 
14
  # run dlc thru list of crops
15
+ dlc_live = DLCLive(dlc_model_folder, processor=dlc_proc)
16
  dlc_live.init_inference(list_np_crops[0])
17
 
18
  list_kpts_per_crop = []