目标检测小脚本:根据xml批量复制jpg图片
Posted zstar-_
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了目标检测小脚本:根据xml批量复制jpg图片相关的知识,希望对你有一定的参考价值。
问题场景
在使用LabelImg标注完目标检测的数据之后,由于很多样本是负例(没有目标的样本),将其输入到模型中意义不大,因此需要进行剔除。
那么实现思路就是根据xml的名称来筛选对应的jpg图片。
文件结构
如图,image是原始图片,xml是标注数据,image_out是筛选后输出的图片。
实现代码
import os
import shutil
if __name__ == '__main__':
xml_src = "xml/"
img_src = "image/"
img_out = "image_out/"
img_list = []
for name in os.listdir(xml_src):
img_list.append(name[:-4])
print(name[:-4])
for i in img_list:
new_obj_name = i + '.jpg'
shutil.copy(img_src + '/' + new_obj_name, img_out)
以上是关于目标检测小脚本:根据xml批量复制jpg图片的主要内容,如果未能解决你的问题,请参考以下文章