charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

make model required

Browse Source
This commit is contained in:
lorenzo 2019-07-26 12:08:49 +02:00
parent 315f401c37
commit cec1ba2c83
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
monoloco/prep/__init__.py
View File

@ -1 +0,0 @@

3
monoloco/run.py
View File

@ -46,8 +46,7 @@ def cli():
predict_parser.add_argument('--model', help='path of MonoLoco model to load',
default="data/models/monoloco-190719-0923.pkl")
predict_parser.add_argument('--hidden_size', type=int, help='Number of hidden units in the model', default=512)
predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization',
default='data/arrays/names-kitti-190726-1141.json')
predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization', required=True)
predict_parser.add_argument('--transform', help='transformation for the pose', default='None')
predict_parser.add_argument('--draw_box', help='to draw box in the images', action='store_true')
predict_parser.add_argument('--predict', help='whether to make prediction', action='store_true')