简单版,python将文件夹内特定类型的图片保存在另一个文件夹里

图像格式
如上图所示:想把00001.png 与00001_matte.png 分存于两个不同的文件夹里,对于大量的数据集单张操作太耗时,简单的文件排序也不能做到,所以我就了几句python代码替代了机械式的操作,话不多说,上代码:

import os 
from PIL import Image
 
path='D:/dataset/training'
dirs=os.listdir(path)  #列表 得到图片名

for i in dirs:
    
        
    if '_' not in os.path.splitext(i)[0]: #os.path.splitext得到的是图片名和扩展名
        print(i)
        img=Image.open(path+'/'+i)
        img.save('D:/data/imgs/'+i)
#        
#    if '_' in os.path.splitext(i)[0]:
##        print(i)
#        img=Image.open(path+'/'+i)
#        
#        img.save('D:/data/masks/'+i)

其中i表示图片的名字,当然可以自行更改,文件路径改成自己的路径就可以了。

你可能感兴趣的:(python,python)