Skip to content
Permalink
Browse files

fall back to config.yaml for num_outputs (#520)

* fall back to config.yaml for num_outputs

* default to config.yaml, and only look within pose_cfg.yaml otherwise
  • Loading branch information
lambdaloop authored and AlexEMG committed Dec 12, 2019
1 parent 1e5b029 commit 2fe0fccd904b95ead21cfb2bfa0591e4ef26f996
Showing with 1 addition and 1 deletion.
  1. +1 −1 deeplabcut/pose_estimation_tensorflow/predict_videos.py
@@ -164,7 +164,7 @@ def analyze_videos(config,videos, videotype='avi', shuffle=1, trainingsetindex=0
dlc_cfg['init_weights'] = os.path.join(modelfolder , 'train', Snapshots[snapshotindex])
trainingsiterations = (dlc_cfg['init_weights'].split(os.sep)[-1]).split('-')[-1]
# Update number of output and batchsize
dlc_cfg['num_outputs'] = dlc_cfg.get('num_outputs', 1)
dlc_cfg['num_outputs'] = cfg.get('num_outputs', dlc_cfg.get('num_outputs', 1))

if batchsize==None:
#update batchsize (based on parameters in config.yaml)

0 comments on commit 2fe0fcc

Please sign in to comment.
You can’t perform that action at this time.