在Google Colab中安装TensorFlow Object Detection API失败求助

问题重现 #

使用过时教程中的代码在Colab（Python 3、T4 GPU）环境安装TensorFlow Object Detection API时，出现metadata生成失败错误，根源为PyYAML依赖冲突。尝试修改PyYAML版本、更新wheel和setuptools均未解决问题。复现错误的代码如下：

!pip install pip --upgrade
%cd /content
!git clone --quiet https://github.com/tensorflow/models.git

%cd /content/models/research/
!protoc object_detection/protos/*.proto --python_out=.
%cp object_detection/packages/tf2/setup.py .
!python -m pip install .

解决方案 #

该依赖冲突已被TensorFlow Models仓库的最新提交修复，只需确保拉取最新代码并按标准流程安装即可：

• 升级pip并克隆最新代码：

!pip install --upgrade pip
%cd /content
!git clone https://github.com/tensorflow/models.git

• 编译Protobuf文件：

%cd /content/models/research/
!protoc object_detection/protos/*.proto --python_out=.

• 安装TF2版本的Object Detection API：

%cp object_detection/packages/tf2/setup.py .
!python -m pip install .

• 可选验证步骤：

from object_detection.utils import label_map_util
from object_detection.utils import visualization_utils as viz_utils
print("TensorFlow Object Detection API安装成功")

内容的提问来源于stack exchange，提问作者Adam Ryason