这次的课后练习是使用从度娘的飞桨paddlepaddle的学习平台上的公开数据集。第一层目录包含train,test,valid个文件夹,每个文件夹内又存有十二种生肖的图像数据。
目录
一,数据集构成
二,数据导入
三,模型搭建
四,模型训练
五,模型测试
一,数据集构成
trian:每种动物600张,共计7200个样本。
valid:每种动物55张,共计660个样本。
test: 每种动物55张,共计660个样本。
二,数据导入
#手动定义数据集目录
train_dir='E:/深度学习数据集/signs/train'
validation_dir='E:/深度学习数据集/signs/valid'
test_dir='E:/深度学习数据集/signs/test'
train_datagen = ImageDataGenerator(rescale=1/255,
rotation_range=40,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
train_generator = train_datagen.flow_from_directory(train_dir, (224, 224), batch_size=100, class_mode='binary', shuffle=True)
test_datagen = ImageDataGenerator(rescale