H5W3
当前位置:H5W3 > 其他技术问题 > 正文

tensorflow训练好的vgg16

想跑一段tensorflow?!想试一试传说中vgg16的神奇?真是太难了。以下是我的一段艰辛过程,阅知即可。

我的主要代码来之与https://blog.csdn.net/helei001/article/details/53159690,将第一个链接下的gitub下好。可以看到关于vgg16的有两个文件 text_vgg16.py和vgg16.py还有一个utils.py文件也是作者这个程序里面要用的,但是我没有实现出来,修改了代码,所以我没有用到,就用了vgg16这个网络的框架文件。

调试初始,就有一个问题困扰着我,就是“找不到vgg16.npy”文件直到我又回去仔细看了gitub才发现,

要先下载好这个文件才行,这是训练好的参数文件,导入后就等于有了参数了。

然后运行text文件时遇到各种各样的问题,索性决定放弃,直接在vgg16网络中进行了修改测试。

主要讲imgs和prob调到了前面以防后面调用找不到,将rgb转gbr还是什么的那块删了,因为不明白为什么要转,而且int类型和float以及tensor类型不同的文件,在原文件的text里面总存在问题,导致调用build失败。索性直接删去也不影响,只是给出的vgg-mean的顺序要调整一下。

测试:

这就是我整个的测试部分,很简单吧,基本是https://blog.csdn.net/u011974639/article/details/76146822里面的测试程序,真的好使。而且里面有详细的vgg原理的讲解,可以看一下。原博实验的结果如下:

但是我采用的是之前下载的vgg16.npy,我也不是很清楚里面的具体情况,我在运行时会出现class_name没有定义,所以就给注释了,有知道的大神欢迎指导。下面是我的运行结果。

就这样。。。。。小白之路不好走啊。。。。。

https://blog.csdn.net/u011974639/article/details/76146822

https://blog.csdn.net/helei001/article/details/53159690

本文地址:H5W3 » tensorflow训练好的vgg16

评论 0

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址