import tensorflow.contrib.slim as slim
ModuleNotFoundError: No module named 'tensorflow.contrib'
上述代码会报错
Traceback (most recent call last):
File "C:/Users/Administrator/Desktop/CV/Faster-RCNN-TensorFlow-Python3-master/demo.py", line 29, in <module>
from lib.nets.vgg16 import vgg16
File "C:\Users\Administrator\Desktop\CV\Faster-RCNN-TensorFlow-Python3-master\lib\nets\vgg16.py", line 9, in <module>
import tensorflow.contrib.slim as slim
ModuleNotFoundError: No module named 'tensorflow.contrib'
原因:tensorflow 2.0以后没有 tensorflow.contrib
解决方案一:降低版本
pip uninstall tensorflow #卸载tensorflow
pip install tensorflow==1.14.0 #安装tensorflow 1.13.1
解决方案二:安装Tf-slim包
pip install tf_slim
代码修改
#import tensorflow.contrib.slim as slim 注释掉原代码,改从tf_slim 中调用slim
import tf_slim as slim
#from tensorflow.contrib.slim import arg_scope 注释掉原代码,改从tf_slim 中调用arg_scope
from tf_slim import arg_scope
TF-Slim是一个轻量级的库,用于在TensorFlow中定义,训练和评估复杂的模型。tf-slim的组件可以与tensorflow以及其他框架自由混合。