在一个小数据集上训练卷积神经网络——猫狗识别

步骤

(1)从kaggle.com网站下载数据： www.kaggle.com/c/dogs-vs-cats/data

(2)制作训练数据集目录

import os, shutil 
original_dataset_dir = './train' 

base_dir = './cats_and_dogs_small'  #当文件已存在时，无法创建该文件
os.mkdir(base_dir) 

train_dir = os.path.join(base_dir, 'train') 
os.mkdir(train_dir) 

validation_dir = os.path.join(base_dir, 'validation') 
os.mkdir(validation_dir) 

test_dir = os.path.join(base_dir, 'test') 
os.mkdir(test_dir)

train_cats_dir = os.path.join(train_dir, 'cats') 
os.mkdir(train_cats_dir) 
train_dogs_dir = os.path.join(train_dir, 'dogs') 
os.mkdir(train_dogs_dir) 

validation_cats_dir = os.path.join(validation_dir, 'cats') 
os.mkdir(validation_cats_dir) 
validation_dogs_dir = os.path.join(validation_dir, 'dogs') 
os.mkdir(validation_dogs_dir) 

test_c