猫狗二分类模型
最新推荐文章于 2022-02-11 17:32:49 发布
本文介绍了一个小型数据集上的猫狗二分类任务,利用2000张训练图片、1000张验证图片和1000张测试图片。通过keras.preprocessing.image的ImageDataGenerator进行预处理和数据增强,使用VGG16模型进行特征提取和微调。未使用数据增强时验证精度为0.9,加入数据增强后提升到0.97,最终微调后的测试精度达0.95。参考书籍《Deep Learning With Python》。
摘要由CSDN通过智能技术生成