前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >python3 下 tensorflow slim inceptionV4 问题修正与测试

python3 下 tensorflow slim inceptionV4 问题修正与测试

作者头像
sparkexpert
发布2018-01-09 11:43:27
9590
发布2018-01-09 11:43:27
举报

自从残差网络出来之后,好多人就想着能不能对模型进行结合测试一下。Google Research的Inception模型和Microsoft Research的ResidualNet模型两大图像识别杀器结合效果如何?在这篇2月23日公布在arxiv上的文章“Inception-v4, Inception-ResNet and the Impactof Residual Connections on Learning”给出了实验上的结论。

发现tensorflow的slim 模型中包含有inception_v4的预训练模型,于是进行测试了下,发现效果相当不错,下面就将测试结果罗列如下:

(1)由于环境用的是python3,发现tensorflow公开的slim模型中的inception存在一些BUG,主要问题在于xrange问题。

  # 35 x 35 x 384

      # 4 x Inception-A blocks

for idx in range(4):

        block_scope ='Mixed_5' + chr(ord('b') + idx)

        net =block_inception_a(net, block_scope)

        ifadd_and_check_final(block_scope, net): return net, end_points

      # 35 x 35 x 384

      # Reduction-A block

      net = block_reduction_a(net,'Mixed_6a')

      ifadd_and_check_final('Mixed_6a', net): return net, end_points

      # 17 x 17 x 1024

      # 7 x Inception-B blocks

   for idxin range(7):

        block_scope ='Mixed_6' + chr(ord('b') + idx)

        net = block_inception_b(net,block_scope)

        ifadd_and_check_final(block_scope, net): return net, end_points

      # 17 x 17 x 1024

      # Reduction-B block

      net =block_reduction_b(net, 'Mixed_7a')

      ifadd_and_check_final('Mixed_7a', net): return net, end_points

      # 8 x 8 x 1536

      # 3 x Inception-C blocks

   for idxin range(3):

        block_scope ='Mixed_7' + chr(ord('b') + idx)

        net =block_inception_c(net, block_scope)

        ifadd_and_check_final(block_scope, net): return net, end_points

  raise ValueError('Unknownfinal endpoint %s' % final_endpoint)

需要把原来代码中的xrange修改为range,这是由于python3与python2环境不兼容问题所导致的。

(2)图像识别测试

识别精度结果:

使用预训练模型:InceptionV4 进行训练后的效果比较:

Probability 85.64% => [tiger, Panthera tigris]

Probability 5.51% => [tiger cat]

Probability 0.19% => [jaguar, panther, Panthera onca, Felis onca]

Probability 0.19% => [lynx, catamount]

Probability 0.08% => [leopard, Panthera pardus]

比vgg直接提高了13个百分点

测试结果如下,也是精度提高了好多。

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2017年04月18日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档