失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > python 图片分别保存至文件夹(深度学习图片数据集处理)

python 图片分别保存至文件夹(深度学习图片数据集处理)

时间:2023-03-23 07:09:44

相关推荐

python 图片分别保存至文件夹(深度学习图片数据集处理)

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 图片分别保存至文件夹(深度学习图片数据集处理)》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。