浏览代码

Update voc.py

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

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

@@ -72,7 +72,7 @@ class VOCDetection(Dataset):
         label_id = 1
         with open(label_list, 'r', encoding=get_encoding(label_list)) as fr:
             for line in fr.readlines():
-                cname2cid[line.strip()]] = label_id
+                cname2cid[line.strip()] = label_id
                 label_id += 1
                 self.labels.append(line.strip())
         logging.info("Starting to read file list from dataset...")