python 图片分别保存至文件夹(深度学习图片数据集处理)
将一个文件夹下的图片进行分类分别保存到两个文件夹中
作为刚接触深度学习的小白,在看了两三个例子之后,开始了自己艰难的探索之路。在网上下载了猫狗识别的数据集,发现猫狗的图片都在一个文件夹里,现在想把猫和狗的图片分别变成两个文件。
参考了博客 /qq_43569111/article/details/100010904 并做了一些修改
import osimport cv2def read_directory(directory_root,objective1_name,objective2_name,class_name):for img_name in os.listdir(directory_root):# print(img_name)img = cv2.imread(directory_root+"/"+img_name)if class_name in img_name.split('/')[-1] : cv2.imwrite(objective1_name+"/"+img_name,img)else:cv2.imwrite(objective2_name + "/" + img_name, img)path = 'F:\dataset\dog_vs_cat/train_o' #原文件路径savepath_1 = 'F:\dataset\dog_vs_cat/train\Dogs'#新文件夹狗的路径savepath_2= 'F:\dataset\dog_vs_cat/train\Cats' #新文件夹猫的路径class_name = 'dog' #其中一类的名字read_directory(path,savepath_1,savepath_2,class_name)
如果觉得《python 图片分别保存至文件夹(深度学习图片数据集处理)》对你有帮助,请点赞、收藏,并留下你的观点哦!