本文试图解决一个问题,即我们自定义的数据如何训练模型?初识深度学习,我们接触手写数字识别模型,但是批次数据是mnist已经定义好的,我们现在有自己的图片如何做成批次进行训练模型. 现在我们将准备好的原始数据放在flower_photos下的文件夹下面,里面又分类包含五种花,文件夹的名字即为类名,每种文件夹下面又有若干图片,如下图所示: 例如,daisy文件下面图片,图片的尺寸大小不一致,这在后期要处理. 数据增强 数据增强通过对图片进行一定的处理,使得训练数据量增强,防止过拟合等,这块后期完善.…