引入vggnet网络时，代码从tensorflow1转换到2时所需要做的import改动与其他网络搜索module的方法

98 阅读 0 评论 65 点赞

我是靠谱客的博主鲜艳书包，最近开发中收集的这篇文章主要介绍引入vggnet网络时，代码从tensorflow1转换到2时所需要做的import改动与其他网络搜索module的方法，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

最常见出现的错误是 module ‘tensorflow’ has no attribute ‘contrib’，这是因为tensorflow2弃用了contrib, 如果刚好对应，可以在引入时使用

import tensorflow.compat.v1 as tf

的方法来解决。

如果不能解决，例如使用slim等模块时仍然报错，那么可以寻找tensorflow中提供的替代方案。
如slim模块的替代方案tf_slim

举例如下：更改前：
在这里插入图片描述
更改后：

这样就可以解决tensorflow2中不能引入contrib.slim的问题了。

如果你遇到的其他模块在引入时的问题，也可以在tensorflow文档中搜索它，确保使用正确的引用路径。如图所示在左上角搜索即可，比如你想用resnet50但是找不到正确的引用路径，就可以在这里搜索它在什么module中。
在这里插入图片描述
附2.8.0版本和1.15版本的网址，比较方便。
https://tensorflow.google.cn/api_docs/python/tf/compat/v1/keras/applications/resnet50
https://tensorflow.google.cn/versions/r1.15/api_docs/python/tf