Tensoflow vgg网络fineturn(微调)

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]

你可能感兴趣的:(Tensorflow,神经网络)