使用python将同一文件下的不同文件分离（将image与label分离)

在训练深度学习数据的时候，有时候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