未验证 提交 a6983cc5 编写于 作者: W Wenyu 提交者: GitHub

voc data format (#3433)

上级 b87a1ea8
...@@ -195,7 +195,8 @@ def voc_get_label_anno(ann_dir_path, ann_ids_path, labels_path): ...@@ -195,7 +195,8 @@ def voc_get_label_anno(ann_dir_path, ann_ids_path, labels_path):
labels_ids = list(range(1, len(labels_str) + 1)) labels_ids = list(range(1, len(labels_str) + 1))
with open(ann_ids_path, 'r') as f: with open(ann_ids_path, 'r') as f:
ann_ids = f.read().split() ann_ids = [lin.strip().split(' ')[-1] for lin in f.readlines()]
ann_paths = [] ann_paths = []
for aid in ann_ids: for aid in ann_ids:
if aid.endswith('xml'): if aid.endswith('xml'):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册