Fix parse error in XMLDataset (open-mmlab#4729)

Pooja0509 · Mar 15, 2021 · f2aa584 · f2aa584
1 parent f3bc86b
commit f2aa584
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmdet/datasets/xml_style.py b/mmdet/datasets/xml_style.py
@@ -102,7 +102,8 @@ def get_ann_info(self, idx):
             if name not in self.CLASSES:
                 continue
             label = self.cat2label[name]
-            difficult = int(obj.find('difficult').text)
+            difficult = obj.find('difficult')
+            difficult = 0 if difficult is None else int(difficult.text)
             bnd_box = obj.find('bndbox')
             # TODO: check whether it is necessary to use int
             # Coordinates may be float type