Avoid saving masks in voc when segmentation is not required (cvat-ai#184

) * Avoid saving masks in voc when segmentation is not required
TOsmanov · Mar 24, 2021 · 6ca7f24 · 6ca7f24
1 parent 7a66b5e
commit 6ca7f24
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -28,6 +28,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Image reading and saving in ICDAR formats (<https://github.com/openvinotoolkit/datumaro/pull/174>)
 - Unnecessary image loading on dataset saving (<https://github.com/openvinotoolkit/datumaro/pull/176>)
 - Allowed spaces in ICDAR captions (<https://github.com/openvinotoolkit/datumaro/pull/182>)
+- Saving of masks in VOC when masks are not requested (<https://github.com/openvinotoolkit/datumaro/pull/184>)
 
 ### Security
 -

diff --git a/datumaro/plugins/voc_format/converter.py b/datumaro/plugins/voc_format/converter.py
@@ -315,7 +315,7 @@ def save_subsets(self):
 
                     clsdet_list[item.id] = True
 
-                if masks:
+                if masks and VocTask.segmentation in self._tasks:
                     compiled_mask = CompiledMask.from_instance_masks(masks,
                         instance_labels=[self._label_id_mapping(m.label)
                             for m in masks])