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

fix force complete pose

Browse Source
This commit is contained in:
Lorenzo 2021-01-06 12:19:08 +01:00
parent e10f012dab
commit 22bc820a9c
2 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monstereo/predict.py
View File

@ -14,6 +14,7 @@ from PIL import Image
import openpifpaf import openpifpaf
import openpifpaf.datasets as datasets import openpifpaf.datasets as datasets
from openpifpaf.predict import processor_factory, preprocess_factory from openpifpaf.predict import processor_factory, preprocess_factory
from openpifpaf import decoder, network, visualizer, show
from .visuals.printer import Printer from .visuals.printer import Printer
from .visuals.pifpaf_show import KeypointPainter, image_canvas from .visuals.pifpaf_show import KeypointPainter, image_canvas
@ -56,6 +57,16 @@ def factory_from_args(args):
else: else:
args.batch_size = 1 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 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) net = Loco(model=args.model, net=args.net, device=args.device, n_dropout=args.n_dropout, p_dropout=args.dropout)
# data # data
processor, model = processor_factory(args) processor, model = processor_factory(args)
preprocess = preprocess_factory(args) preprocess = preprocess_factory(args)

17
monstereo/run.py
View File

@ -1,11 +1,8 @@
# pylint: disable=too-many-branches, too-many-statements # pylint: disable=too-many-branches, too-many-statements
import argparse import argparse
try:
from openpifpaf.network.nets import cli as openpifpaf_cli from openpifpaf import decoder, network, visualizer, show
except ImportError:
from openpifpaf.network.factory import cli as openpifpaf_cli
from openpifpaf import decoder
def cli(): def cli():
@ -39,13 +36,13 @@ def cli():
help='what to output: json keypoints skeleton for Pifpaf' help='what to output: json keypoints skeleton for Pifpaf'
'json bird front or multi for MonStereo') 'json bird front or multi for MonStereo')
predict_parser.add_argument('--no_save', help='to show images', action='store_true') 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('--dpi', help='image resolution', type=int, default=100)
predict_parser.add_argument('--force-complete-pose', help='', action ='store_true')
# Pifpaf # Pifpaf parsers
openpifpaf_cli(predict_parser)
decoder.cli(predict_parser) 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') predict_parser.add_argument('--scale', default=1.0, type=float, help='change the scale of the image to preprocess')
# Monoloco # Monoloco
@ -55,7 +52,7 @@ def cli():
predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization', predict_parser.add_argument('--path_gt', help='path of json file with gt 3d localization',
default='data/arrays/names-kitti-200615-1022.json') default='data/arrays/names-kitti-200615-1022.json')
predict_parser.add_argument('--transform', help='transformation for the pose', default='None') 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('--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('--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') predict_parser.add_argument('--show_all', help='only predict ground-truth matches or all', action='store_true')