AttributeError: module ‘tensorflow’ has no attribute ‘contrib’;Tensorflow2x版本使用slim()问题;python setup.py install步骤
在复现tensorflow框架下远古的一个分类项目时候,出现以下报错:
AttributeError: module ‘tensorflow’ has no attribute 'contrib’
Traceback (most recent call last):
File "E:/Python/python_user/Practice_codes/second/train.py", line 7, in <module>
slim = tf.contrib.slim
AttributeError: module 'tensorflow' has no attribute 'contrib'
对应源码导入第三方库部分为:
import tensorflow as tf
import tf_slim as slim
slim = tf.contrib.slim
后来得知tensorflow2.0版本以后取消了contrib库,但是项目源码里有大量slim()函数。当时有两条路摆在我面前:1.大修模型框架;2.解决slim()问题,在大修模型五分钟后,1个bug变成了4个,然后…
Tensorflow2x版本使用slim()问题
踏上了第二条路后,查阅相关博客,尝试了以下操作:
import tensorflow.compat.v1 as tf
然并卵~~~
后来就是大量查找相关博客、百度、知乎、简书、博客园,10分钟、20分钟、30分钟、40分钟过去后,终于!我看到了TensorFlow-Slim image classification model library,然后就根据指引下载了Slim的一个版本:Note: Latest version of TF-Slim, 1.1.0, was tested with TF 1.15.2 py2, TF 2.0.1, TF 2.1 and TF 2.2..。
python setup.py install步骤
下载之后得到了setup.py文件,接下来就是如何在tensorflow环境中安装第三方的Slim1.1,步骤如下:
- 从GitHub上下载安装包解压,不会下载解压的小伙伴请长按电源键;
- 将解压后的文件夹放入tensorflow环境中,如下:
- 打开cmd命令窗口,不会的小伙伴们再次长按电源键;
- 在cmd中进入文件指定位置,然后输入
python setup.py install
经个人实验,不需要先运行python setup.py bulid 然后再运行 python setup.py install,因为第二步包含第一步。
- 等待,等待,然后关闭;如下:
之后在原程序中修改代码为:
import tensorflow.compat.v1 as tf
import tf_slim as slim
然后就OK!
给出几篇 Nice 的参考博客:
python:用setup.py安装第三方包packages
python安装模块如何通过setup.py安装
简述python setup.py install的过程
文章评论