在训练深度学习数据的时候,有时候image图片数据和XML文件混在了一起。使用下面的程序将其分开。
import os import shutil root ="/home/jk/Downloads/yolov5/xiamen_60_day" #需要分离的文件,即image 与xml在此文件下 image ="/home/jk/Downloads/yolov5/image/" label ="/home/jk/Downloads/yolov5/label " for x in os.scandir(root): if x.name.endswith(".jpg"): needpath = "/home/jk/Downloads/yolov5/xiamen_60_day/{name}".format(name=x.name) shutil.move(needpath,image) if x.name.endswith(".xml"): needpath ="/home/jk/Downloads/yolov5/xiamen_60_day/{name}".format(name=x.name) shutil.move(needpath,label)
代写程序,代接各种项目,算法,软硬件均可接
联系vx :18811396527