VGG16 pre-trained model 实现image classification

Posted 暴力的轮胎

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了VGG16 pre-trained model 实现image classification相关的知识,希望对你有一定的参考价值。

 站在巨人的肩膀上!使用VGG预先训练好的weight来,进行自己的分类。

 下一阶段是在这上面进行自己的修改,完成自己想要的功能。

Github源码

Github上有我全部的工程代码。

环境配置

  Python3.5
  Keras2.0
  TensorFlow
  我这里是在Windows10, LINUX一样配置对应的package就好了,记得路径改了就ok

关于各种错误:

  github glist中有我的回答,几乎包括了所有可能发生的错误(我几乎碰到了所有可能的错误,囧)。

  直接搜错误log 或 搜我的id : luntai就可以找到。

测试

n02123045 tabby, tabby cat

 

 

n02391049 zebra

  一个模型搞了一整天,跪在坚持,贵在坚持。陪这种东西很虐心 但是孰能手巧。慢慢就会找到感觉了。

以上是关于VGG16 pre-trained model 实现image classification的主要内容,如果未能解决你的问题,请参考以下文章

Keras:什么是VGG16中的model.inputs

pytorch笔记:VGG 16

Keras VGG16 微调

FailedPreconditionError:尝试使用未初始化的值Adam / lr

带有 VGG16 模型拟合误差的迁移学习

从 Vgg16 网络导入后如何更改瓶颈特征的输入形状