深度学习实战---猫狗大战(pytorch实现)

最新推荐文章于 2025-10-24 18:30:03 发布

原创

最新推荐文章于 2025-10-24 18:30:03 发布 · 1.1w 阅读

98 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #pytorch

数据准备

猫狗大战数据集下载链接

微软的数据集已经分好类，直接使用就行，

数据划分

我们将猫和狗的图片分别移动到训练集和验证集中，其中90%的数据作为训练集，10%的图片作为验证集，使用shutil.move（）来移动图片。

新建文件夹train,test,将数据集放入train中，利用代码将10%的数据移动到test中

文件移动代码

import os
import shutil
source_path = r"E:\猫狗大战数据集\PetImages"
train_dir = os.path.join(source_path, "train")
test_dir = os.path.join(source_path,"test")
train_dir_list = os.listdir(train_dir)
for dir in train_dir_list:
    category_dir_path = os.path.join(train_dir, dir)
    image_file_list = os.listdir(category_dir_path)
    num = int(0.1*len(image_file_list))

    #移动10%文件到对应目录
    for i in range(num):
        shutil.move(os.path.join(category_dir_path,image_file_list[i]),os.path.join(test_dir,dir,image_