From 8dd34643bc1385bdd7eee6de4246510c38bd8196 Mon Sep 17 00:00:00 2001
From: Edward Yang <eepgxxy@163.com>
Date: Mon, 28 Mar 2022 19:54:52 +0800
Subject: [PATCH] Update voc.py (#5407)

Updated voc.py in case error occurs when xml file contains self closing node <difficult/>.
---
 ppdet/data/source/voc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/data/source/voc.py b/ppdet/data/source/voc.py
index 1c2a7ef98..fb58f3a4d 100644
--- a/ppdet/data/source/voc.py
+++ b/ppdet/data/source/voc.py
@@ -143,7 +143,7 @@ class VOCDataSet(DetDataset):
                     # user dataset may not contain difficult field
                     _difficult = obj.find('difficult')
                     _difficult = int(
-                        _difficult.text) if _difficult is not None else 0
+                        _difficult.text) if _difficult is not None and _difficult.text is not None else 0
 
                     x1 = float(obj.find('bndbox').find('xmin').text)
                     y1 = float(obj.find('bndbox').find('ymin').text)
-- 
GitLab