如何在苹果硅(ARM/M1)设备上安装TensorFlow 2 Object Detection API

前置环境校验 #

• 确认当前使用原生arm64架构终端，执行arch命令返回arm64为符合要求，请勿使用Rosetta 2转译的终端
• 确认已安装的TensorFlow 2适配Apple Silicon架构，执行验证命令无报错即可正常使用：
  python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

依赖项安装 #

执行以下命令安装所需依赖：

conda install protobuf=3.20.3 pillow matplotlib lxml opencv
pip install cython pycocotools

源码编译与安装 #

• 获取TensorFlow Models源码到本地目录，进入models/research路径
• 编译所有Protobuf文件：
  protoc object_detection/protos/*.proto --python_out=.
• 复制安装配置文件并执行安装：
  cp object_detection/packages/tf2/setup.py .
pip install .

安装有效性校验 #

执行测试命令，返回OK即代表安装成功：
python object_detection/builders/model_builder_tf2_test.py

常见故障处理 #

• 若出现Protobuf版本兼容报错，强制安装3.20.x版本即可解决：pip install protobuf==3.20.3
• 若出现包架构不兼容报错，优先通过conda渠道安装对应包，不要直接使用pip安装

内容的提问来源于stack exchange，提问作者Nipa Das Gupta