未验证 提交 5d59995c 编写于 作者: S SunAhong1993 提交者: GitHub

Update voc.py

上级 b0b089b8
...@@ -72,7 +72,7 @@ class VOCDetection(Dataset): ...@@ -72,7 +72,7 @@ class VOCDetection(Dataset):
label_id = 1 label_id = 1
with open(label_list, 'r', encoding=get_encoding(label_list)) as fr: with open(label_list, 'r', encoding=get_encoding(label_list)) as fr:
for line in fr.readlines(): for line in fr.readlines():
cname2cid[line.strip()]] = label_id cname2cid[line.strip()] = label_id
label_id += 1 label_id += 1
self.labels.append(line.strip()) self.labels.append(line.strip())
logging.info("Starting to read file list from dataset...") logging.info("Starting to read file list from dataset...")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册