diff --git a/monoloco/run.py b/monoloco/run.py index 1d1b8bf..a9c3236 100644 --- a/monoloco/run.py +++ b/monoloco/run.py @@ -43,10 +43,10 @@ def cli(): predict_parser.add_argument('--scale', default=1.0, type=float, help='change the scale of the image to preprocess') # Monoloco - predict_parser.add_argument('--model', help='path of MonoLoco model to load', - default="data/models/monoloco-190719-0923.pkl") + predict_parser.add_argument('--model', help='path of MonoLoco model to load', required=True) 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', required=True) + predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization', + default='data/arrays/names-kitti-190513-1754.json') 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')