【下载数据集】

• 下载链接–百度网盘
  

【整理数据集】

• 将训练数据集分割成训练集、验证集、测试集，目录结构如图所示：
  

• 在Pycharm中新建项目，创建split_dataset.py

import os, shutil# 数据集解压之后的目录
original_dataset_dir = 'D:\kaggle\dogsvscats\\train'
# 存放小数据集的目录
base_dir = 'D:\kaggle\dogsvscats\\cats_and_dogs_small'
os.mkdir(base_dir)# 建立训练集、验证集、测试集