Skip to content

Commit 97288a8

Browse files
author
--global
committed
make nuscenes model default
1 parent 889ac5d commit 97288a8

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

monoloco/predict.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff 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, \

0 commit comments

Comments
 (0)