使用对象检测API

问题描述

我正在尝试使用Tensorflow 2在https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md中训练ssd_inception_v2_coco_2018_01_28,但出现错误 ValueError:不支持ssd_inception_v2。有关与不同版本的Tensorflow兼容的特征提取器,请参见model_builder.py。 我尝试使用tensorflow 1.15创建一个新环境,并使用object_detection / packages / tf1 / setup.py在其上安装了对象检测api,认为这可以解决问题,因为我为tensorflow 1安装了对象检测api(或者我相信),但是当试图训练我得到同样的错误。我正在使用文件model_main.py,model_main_tf2.py和旧版train.py,但无济于事。我会认为即时通讯使用错误文件来训练tensorflow 1模型,但我找不到关于它的任何文档。任何帮助将不胜感激!

解决方法

请转到tf1 installation并遵循 Python软件包安装。一旦成功运行以下测试,您就可以在系统上使用tensorflow对象检测api。

# Test the installation.
python object_detection/builders/model_builder_tf1_test.py

使用tf1_detection_zoo模型,如果仍然遇到任何错误,可以共享它。