From e29ee4d75bfc1daf86f015f879b233103e19b596 Mon Sep 17 00:00:00 2001 From: sunyanfang01 Date: Fri, 22 May 2020 11:12:34 +0800 Subject: [PATCH] fix the voc dataset --- paddlex/cv/datasets/voc.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/paddlex/cv/datasets/voc.py b/paddlex/cv/datasets/voc.py index 820a206..fee0c78 100644 --- a/paddlex/cv/datasets/voc.py +++ b/paddlex/cv/datasets/voc.py @@ -133,17 +133,17 @@ class VOCDetection(Dataset): box_tag = pattern.findall(str(ET.tostringlist(obj)))[0][1:-1] box_element = obj.find(box_tag) pattern = re.compile('', re.IGNORECASE) - xmin_element = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] - x1 = float(box_element.find(xmin_element).text) + xmin_tag = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] + x1 = float(box_element.find(xmin_tag).text) pattern = re.compile('', re.IGNORECASE) - ymin_element = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] - y1 = float(box_element.find(ymin_element).text) + ymin_tag = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] + y1 = float(box_element.find(ymin_tag).text) pattern = re.compile('', re.IGNORECASE) - xmax_element = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] - x2 = float(box_element.find(xmax_element).text) + xmax_tag = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] + x2 = float(box_element.find(xmax_tag).text) pattern = re.compile('', re.IGNORECASE) - ymax_element = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] - y2 = float(box_element.find(ymax_element).text) + ymax_tag = pattern.findall(str(ET.tostringlist(box_element)))[0][1:-1] + y2 = float(box_element.find(ymax_tag).text) x1 = max(0, x1) y1 = max(0, y1) if im_w > 0.5 and im_h > 0.5: -- GitLab