remove check for 'difficult' attribute

openvinotoolkit · Apr 12, 2021 · 53270dd · 53270dd
1 parent 989f990
commit 53270dd
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/datumaro/plugins/voc_format/converter.py b/datumaro/plugins/voc_format/converter.py
@@ -241,15 +241,14 @@ def save_subsets(self):
                             ET.SubElement(obj_elem, 'truncated').text = \
                                 '%d' % truncated
 
-                        if 'difficult' in attr:
-                            difficult = _convert_attr('difficult', attr, int, 0)
-                            ET.SubElement(obj_elem, 'difficult').text = \
-                                '%d' % difficult
-
                         if 'occluded' in attr:
                             occluded = _convert_attr('occluded', attr, int, 0)
                             ET.SubElement(obj_elem, 'occluded').text = \
                                 '%d' % occluded
+
+                        difficult = _convert_attr('difficult', attr, int, 0)
+                        ET.SubElement(obj_elem, 'difficult').text = \
+                            '%d' % difficult
 
                         bbox = obj.get_bbox()
                         if bbox is not None: