网站建设分享,网站建设 引导,wordpress加密授权,唯品会网站建设建议介绍
在目标检测中#xff0c;有些时候会遇到标注好的类别不想要了的情况#xff0c;这时我们可以运行下面的代码来批量删除不需要的类别节省时间。
代码实现#xff1a; import argparseimport xml.etree.ElementTree as ET
import osclasses [thin_smoke]def GetImgNam…介绍
在目标检测中有些时候会遇到标注好的类别不想要了的情况这时我们可以运行下面的代码来批量删除不需要的类别节省时间。
代码实现 import argparseimport xml.etree.ElementTree as ET
import osclasses [thin_smoke]def GetImgNameByEveryDir(file_dir, videoProperty):FileNameWithPath, FileName, FileDir [], [], []for root, dirs, files in os.walk(file_dir):for file in files:if os.path.splitext(file)[1] in videoProperty:FileNameWithPath.append(os.path.join(root, file)) # 保存图片路径FileName.append(file) # 保存图片名称FileDir.append(root[len(file_dir):]) # 保存图片所在文件夹return FileName, FileNameWithPath, FileDirdef GetBoxInfo(xmlfile):try:tree ET.parse(xmlfile)root tree.getroot()size root.find(size)w int(size.find(width).text)h int(size.find(height).text)except:return False, 0else:tree ET.parse(xmlfile)root tree.getroot()size root.find(size)w int(size.find(width).text)h int(size.find(height).text)for obj in root.findall(object):print(xmlfile, obj.find(name).text, obj)if obj.find(name).text fulll:obj.find(name).text fullprint(- * 66)tree.write(xmlfile)return 1def Process(ProcessDir):xmlDirs ProcessDir # Labels/FileName1, FileNameWithPath1, FileDir1 GetImgNameByEveryDir(xmlDirs, .xml)for k in range(len(FileName1)):annfile xmlDirs FileName1[k][:-4] .xmlresult GetBoxInfo(annfile)if __name__ __main__:parser argparse.ArgumentParser()parser.add_argument(--ProcessDir, typestr, default/home/build/smoke_ori/VOCdevkit/VOC2007/Annotations/)args parser.parse_args()ProcessDir args.ProcessDirProcess(ProcessDir)
