From f2aa584b96b255a6812960421e5fef1af00d3787 Mon Sep 17 00:00:00 2001
From: Ye Liu <yeliudev@outlook.com>
Date: Mon, 15 Mar 2021 23:45:05 +0800
Subject: [PATCH] Fix parse error in XMLDataset (#4729)
---
mmdet/datasets/xml_style.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/mmdet/datasets/xml_style.py b/mmdet/datasets/xml_style.py
index 0a0a4b3f..71069488 100644
--- a/mmdet/datasets/xml_style.py
+++ b/mmdet/datasets/xml_style.py
@@ -102,7 +102,8 @@ class XMLDataset(CustomDataset):
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
--
GitLab