TensorFlow 2.x调用tf.contrib.slim报AttributeError，求无需降级TF1.x的方案

问题背景 #

在Jupyter Notebook运行自定义目标检测模型，使用TensorFlow版本≤2.15.0时，出现以下错误：

AttributeError: module 'tensorflow' has no attribute 'contrib'

错误源于代码中尝试导入slim = tf.contrib.slim，而TensorFlow 2.x已彻底移除了tf.contrib模块。

无需切换到TF1.x的解决方案 #

1. 安装独立的TensorFlow Slim库 #

TensorFlow 2.x将slim从contrib中拆分出来，作为独立的tensorflow-slim包发布：

• 先安装包：

  pip install tensorflow-slim
  

• 修改代码中的导入语句，替换原来的tf.contrib.slim：

  import functools
  import tensorflow as tf
  from object_detection.core import box_predictor
  from object_detection.utils import static_shape
  
  # 替换为独立slim库的导入
  import tensorflow_slim as slim
  

2. 确保Object Detection API为TF2兼容版本 #

旧版本的Object Detection API可能仍依赖TF1.x的contrib模块，需要更新到适配TF2的版本：

• 如果是通过pip安装的API，执行升级：

  pip install --upgrade object-detection
  

• 如果是从GitHub克隆的models仓库，切换到tf2分支并重新安装：

  cd models/research
  git checkout tf2
  pip install .
  

3. 用TF2原生API替代slim调用（可选） #

如果部分slim操作仍存在兼容性问题，可以直接替换为TensorFlow 2.x的原生API：

• 例如将slim.conv2d替换为tf.keras.layers.Conv2D，slim.fully_connected替换为tf.keras.layers.Dense，根据具体代码逻辑调整。

内容的提问来源于stack exchange，提问作者Vinay Samuel