python 图片分别保存至文件夹（深度学习图片数据集处理）_如何使用python代码把一个文件夹里的猫狗照片分成两个文件夹

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

将一个文件夹下的图片进行分类分别保存到两个文件夹中

作为刚接触深度学习的小白，在看了两三个例子之后，开始了自己艰难的探索之路。在网上下载了猫狗识别的数据集，发现猫狗的图片都在一个文件夹里，现在想把猫和狗的图片分别变成两个文件。
参考了博客 https://blog.csdn.net/qq_43569111/article/details/100010904 并做了一些修改

import os
import cv2

def 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)       
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17