Tensoflow vgg网络fineturn(微调)
github地址:vgg-fineturn
微调是只训练vgg最后的全连接层,可以在很短的时间让模型收敛,不需要很大的数据集就可以让准确度变得很高。
预先准备:
- python3
- tensorflow
- vgg16.npy文件官方地址, 百度云下载 百度云密码:4wvq
- 将vgg16.npy文件放入vgg_finturning目录下
- 将想要训练的文件放到vgg_finturning目录下,子目录为分类文件,文件名为分类名。eg:birds
代码说明:
1, 训练代码vgg16train.py
python3 vgg16train.py [文件名eg:birds]
2, 测试代码vgg16predict.py
python3 vgg16predict.py [文件名eg:birds] [图片路径eg:birds/owl/owl001.jpg]