总目标:建立VOC格式数据集类以及操作函数库
VOC类库存储在VOC.py中,单个xml文件操作函数库存储在VOCOperationLibrary.py
VOC类import VOCOperationLibrary as vol
github项目地址(附有使用说明书):
https://github.com/A-mockingbird/VOCtype-datasetOperation
Day6.获取指定类别的全部xml文件和图像文件,并复制另存为
1.遍历全部标注文件,对每一个标注文件进行查询
cls:字符串数组,存储要查找的类别名,例如['dog', 'cat']
annodir:标注文件目录
"""
VOC.py,class VOC(object)
import VOCOperationLibrary as vol
"""
def _Find(self, cls, annodir=None):
if annodir == None:
annodir = self.dataset_anno
annolist = self._listanno(annodir)
xml_files = []
#遍历全部标注文件,对每一个标注文件查询
for anno in annolist:
xml = vol._find_one(os.path.join(annodir, anno), cls)