猫狗二分类模型点击这里。 记录几点: (1)这是一个在小型数据集上训练模型的案例(训练集2000张图片,验证集1000张,测试集1000张)。图片资源点击这里。 (2)图像输入之前需要预处理,将其转换为浮点数张量,使用了keras.preprocessing.image中的ImageDataGenerator类,它可以实现图像的缩放,旋转,平移,错切等操作。 展示几张变换的图片: