File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed
Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -77,14 +77,15 @@ def download_checkpoints(args):
7777 assert 'social_distance' not in args .activities , "Social distance not supported in stereo modality"
7878 path = MONSTEREO_MODEL
7979 name = 'monstereo-201202-1212.pkl'
80- elif ('social_distance' in args .activities ) or args .webcam :
81- path = MONOLOCO_MODEL_NU
82- name = 'monoloco_pp-201207-1350.pkl'
83- else :
80+ elif args .calibration == 'kitti' or args .path_gt is not None :
8481 path = MONOLOCO_MODEL_KI
8582 name = 'monoloco_pp-201203-1424.pkl'
83+ else :
84+ path = MONOLOCO_MODEL_NU
85+ name = 'monoloco_pp-201207-1350.pkl'
8686 model = os .path .join (torch_dir , name )
8787 dic_models [args .mode ] = model
88+
8889 if not os .path .exists (model ):
8990 os .makedirs (torch_dir , exist_ok = True )
9091 assert DOWNLOAD is not None , \
You can’t perform that action at this time.
0 commit comments