From 78f1893ab262fe21b666c1c9cd325a1bdbff02fc Mon Sep 17 00:00:00 2001 From: zhiltsov-max Date: Fri, 10 Jul 2020 22:27:43 +0300 Subject: [PATCH] Allow saving empty xmls in voc (#1883) --- datumaro/datumaro/plugins/voc_format/converter.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/datumaro/datumaro/plugins/voc_format/converter.py b/datumaro/datumaro/plugins/voc_format/converter.py index 1465a06a..67130a78 100644 --- a/datumaro/datumaro/plugins/voc_format/converter.py +++ b/datumaro/datumaro/plugins/voc_format/converter.py @@ -20,8 +20,7 @@ from datumaro.util import str_to_bool, find from datumaro.util.image import save_image from datumaro.util.mask_tools import paint_mask, remap_mask -from .format import (VocTask, VocPath, - VocInstColormap, VocPose, +from .format import (VocTask, VocPath, VocInstColormap, parse_label_map, make_voc_label_map, make_voc_categories, write_label_map ) @@ -156,7 +155,9 @@ class _Converter: elif a.type == AnnotationType.mask: masks.append(a) - if len(bboxes) != 0: + if self._tasks is None and bboxes or \ + self._tasks & {VocTask.detection, VocTask.person_layout, + VocTask.action_classification}: root_elem = ET.Element('annotation') if '_' in item.id: folder = item.id[ : item.id.find('_')]