10 monkey | CNN | Tensorflow | Kaggle|迁移学习
迁移学习读取图片和其标签,保存成模型读取的格式模型定义仅训练ResRet最后一层的模型训练最后几层的模型训练网络
读取图片和其标签,保存成模型读取的格式
前面代码和上一篇文章一样
# 做卷积的时候所有图片的尺寸应该是一样的
height = 224
width = 224
channels = 3
batch_size = 24
num_classes = 10
# 读取训练数据并作数据增强
# 确定一些读取格式要求
train_datagen = keras.preprocessing.image.I
原创
2020-08-14 15:46:09 ·
216 阅读 ·
0 评论