如上图所示:想把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表示图片的名字,当然可以自行更改,文件路径改成自己的路径就可以了。