From f929b4b184796d3522b77e3e2fde1d18426bf0ef Mon Sep 17 00:00:00 2001 From: Zhiltsov Max Date: Tue, 23 Jun 2020 11:09:21 +0300 Subject: [PATCH] update coco cli --- .../datumaro/plugins/coco_format/converter.py | 21 ++++++++++--------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/datumaro/datumaro/plugins/coco_format/converter.py b/datumaro/datumaro/plugins/coco_format/converter.py index f3f125b6e24a..0e7dd3a444c1 100644 --- a/datumaro/datumaro/plugins/coco_format/converter.py +++ b/datumaro/datumaro/plugins/coco_format/converter.py @@ -562,22 +562,23 @@ def _split_tasks_string(s): @classmethod def build_cmdline_parser(cls, **kwargs): + kwargs['description'] = """ + Segmentation save modes:|n + - '{sm.guess.name}': guess the mode for each instance,|n + |s|suse 'is_crowd' attribute as a hint|n + - '{sm.polygons.name}': save polygons,|n + |s|smerge and convert masks, prefer polygons|n + - '{sm.mask.name}': save masks,|n + |s|smerge and convert polygons, prefer masks + """.format(sm=SegmentationMode) parser = super().build_cmdline_parser(**kwargs) + parser.add_argument('--save-images', action='store_true', help="Save images (default: %(default)s)") parser.add_argument('--segmentation-mode', choices=[m.name for m in SegmentationMode], default=SegmentationMode.guess.name, - help=""" - Save mode for instance segmentation:|n - - '{sm.guess.name}': guess the mode for each instance,|n - |s|suse 'is_crowd' attribute as hint|n - - '{sm.polygons.name}': save polygons,|n - |s|smerge and convert masks, prefer polygons|n - - '{sm.mask.name}': save masks,|n - |s|smerge and convert polygons, prefer masks|n - Default: %(default)s. - """.format(sm=SegmentationMode)) + help="Save mode for instance segmentation (default: %(default)s)") parser.add_argument('--crop-covered', action='store_true', help="Crop covered segments so that background objects' " "segmentation was more accurate (default: %(default)s)")