使用 Tensorflow 版本 1.14.0 进行自定义对象检测时出现 ImportError

Question

我正在尝试使用 TensorFlow 检测对象，如 here 所述。

在训练部分运行此命令后

python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_pets.config

我收到以下错误：

ImportError: cannot import name 'center_net_pb2' from 'object_detection.protos' (E:\Projects\Tensorflow_10Nov20\models-master\research\object_detection\protos\__init__.py)

这里是相关代码

from object_detection.protos import center_net_pb2 as object__detection_dot_protos_dot_center__net__pb2

Answer 1

不知道您的 protoc 版本是什么，但尝试将其升级到至少 3.7.x 版本，这可能会解决问题。