浏览代码

fix the vo

sunyanfang01 5 年之前
父节点
当前提交
ca3a8bf1b0
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/datasets/voc.py

+ 1 - 1
paddlex/cv/datasets/voc.py

@@ -113,7 +113,7 @@ class VOCDetection(Dataset):
                 is_crowd = np.zeros((len(objs), 1), dtype=np.int32)
                 difficult = np.zeros((len(objs), 1), dtype=np.int32)
                 for i, obj in enumerate(objs):
-                    cname = obj.find('name').text
+                    cname = obj.find('name').text.strip()
                     gt_class[i][0] = cname2cid[cname]
                     _difficult = int(obj.find('difficult').text)
                     x1 = float(obj.find('bndbox').find('xmin').text)