From 22bc820a9c843eb75b27781b3104dba2634eb957 Mon Sep 17 00:00:00 2001 From: Lorenzo Date: Wed, 6 Jan 2021 12:19:08 +0100 Subject: [PATCH] fix force complete pose --- monstereo/predict.py | 12 +++++++++++- monstereo/run.py | 17 +++++++---------- 2 files changed, 18 insertions(+), 11 deletions(-) diff --git a/monstereo/predict.py b/monstereo/predict.py index 416dadf..9984f58 100644 --- a/monstereo/predict.py +++ b/monstereo/predict.py @@ -14,6 +14,7 @@ from PIL import Image import openpifpaf import openpifpaf.datasets as datasets from openpifpaf.predict import processor_factory, preprocess_factory +from openpifpaf import decoder, network, visualizer, show from .visuals.printer import Printer from .visuals.pifpaf_show import KeypointPainter, image_canvas @@ -56,6 +57,16 @@ def factory_from_args(args): else: args.batch_size = 1 + # Make default pifpaf argument + args.force_complete_pose = True + args.instance_threshold = 0.15 + + # Configure + decoder.configure(args) + network.configure(args) + show.configure(args) + visualizer.configure(args) + return args @@ -71,7 +82,6 @@ def predict(args): net = Loco(model=args.model, net=args.net, device=args.device, n_dropout=args.n_dropout, p_dropout=args.dropout) # data - processor, model = processor_factory(args) preprocess = preprocess_factory(args) diff --git a/monstereo/run.py b/monstereo/run.py index 7d0ef06..7215b5c 100644 --- a/monstereo/run.py +++ b/monstereo/run.py @@ -1,11 +1,8 @@ # pylint: disable=too-many-branches, too-many-statements import argparse -try: - from openpifpaf.network.nets import cli as openpifpaf_cli -except ImportError: - from openpifpaf.network.factory import cli as openpifpaf_cli -from openpifpaf import decoder + +from openpifpaf import decoder, network, visualizer, show def cli(): @@ -39,13 +36,13 @@ def cli(): help='what to output: json keypoints skeleton for Pifpaf' 'json bird front or multi for MonStereo') predict_parser.add_argument('--no_save', help='to show images', action='store_true') - predict_parser.add_argument('--show', help='to show images', action='store_true') predict_parser.add_argument('--dpi', help='image resolution', type=int, default=100) - predict_parser.add_argument('--force-complete-pose', help='', action ='store_true') - # Pifpaf - openpifpaf_cli(predict_parser) + # Pifpaf parsers decoder.cli(predict_parser) + network.cli(predict_parser) + show.cli(predict_parser) + visualizer.cli(predict_parser) predict_parser.add_argument('--scale', default=1.0, type=float, help='change the scale of the image to preprocess') # Monoloco @@ -55,7 +52,7 @@ def cli(): predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization', default='data/arrays/names-kitti-200615-1022.json') predict_parser.add_argument('--transform', help='transformation for the pose', default='None') - predict_parser.add_argument('--z_max', type=int, help='maximum meters distance for predictions', default=100) + predict_parser.add_argument('--z_max', type=int, help='maximum meters distance for predictions', default=30) predict_parser.add_argument('--n_dropout', type=int, help='Epistemic uncertainty evaluation', default=0) predict_parser.add_argument('--dropout', type=float, help='dropout parameter', default=0.2) predict_parser.add_argument('--show_all', help='only predict ground-truth matches or all', action='store_true')