件太大(1.05G)需要百度网盘下载。
所有代码用Keras实现,覆盖猫狗识别的基础网络、数据增强、VGG网络特征提取以及模型微调。
data文件夹是要用到的Kaggle数据集
1.DATA.py 用于Kaggle准备数据集,输出结果在data文件夹下
2.data_augmentation.py 用于展示数据增强的效果
3.Model.py 是个简单的识别猫狗的卷积+全连接的网络
4.Model_Aug.py 在 Model.py 的基础上加了数据增强
5.Model_VGG.py 使用VGG作为基础网络,只训练分类器
6.Model_VGG_Aug.py 在 Model_VGG.py 上加了数据增强
7.Model_VGG_fine_tuning.py 在 Model_VGG_Aug 的基础上对VGG网络的block5进行微调。
从3-4-5-6-7可以发现,网络的精度不断提升
1