混合文件下分离xml文件和jpg文件
生活随笔
收集整理的這篇文章主要介紹了
混合文件下分离xml文件和jpg文件
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
import os
import shutildef split(src,img,annota):imges=[] #存儲所有圖片的路徑annotation=[] #存儲所有xml的路徑#第一步:遍歷需要分離的文件夾for f in os.listdir(src):if f.endswith(".jpg"):imges.append(f)if f.endswith(".jpeg"):imges.append(f)if f.endswith(".png"):imges.append(f)if f.endswith(".xml"):annotation.append(f)#第二步:創建目標圖片文件夾和xml文件夾if not os.path.isdir(img): #如果目標圖片文件夾不存在os.mkdir(img)if not os.path.isdir(annota): #如果目標xml文件夾不存在os.mkdir(annota)#第三步:轉移到目標文件夾中for im in imges: #遍歷所有的圖片,將圖片文件轉移到目標文件夾中new_path=os.path.join(src,im)print(new_path)shutil.copy(new_path,img)for ann in annotation: #遍歷所有的xml,將xml文件轉移到目標文件夾中new_path=os.path.join(src,ann)print(new_path)shutil.copy(new_path,annota)if __name__=="__main__":point="./" #相對路徑SrcDir="train" #需要分離的文件dicimg="images" #目標圖片文件dicann="annotations" #目標的xml文件src=os.path.join(point,SrcDir);imges=os.path.join(point,dicimg)annoations=os.path.join(point,dicann)split(src,imges,annoations)
總結
以上是生活随笔為你收集整理的混合文件下分离xml文件和jpg文件的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 初学Jedis
- 下一篇: BM39 序列化二叉树