在Tensorflow版本1.14和2.0上训练Tensorflow对象检测模型时出错

Question

我正在尝试训练Tensorflow对象检测模型。在训练中，当我运行最终命令时：

对于tensorflow版本1.14：python model_main.py --logtostderr --train_dir="/path to dir training/" --pipeline_config_path="Path to ssd_mobilenet_v1_pets_Colab.config"

对于tensorflow 2.0版：python model_main_tf2.py --logtostderr --train_dir="/path to dir training/" --pipeline_config_path="Path to ssd_mobilenet_v1_pets_Colab.config"

在第一种情况下（tensorflow 1.14），我收到以下错误消息：ValueError: ssd_mobilenet_v1 is not supported. See model_builder.py for features extractors compatible with different versions of Tensorflow

在第二种情况（tensorflow 2.0）中，我得到ValueError: ssd_mobilenet_v1 is not supported. See model_builder.py for features extractors compatible with different versions of Tensorflow

我尝试了所有可能的方法，但不清楚应该怎么做。 附言我是深度学习的新手。

Answer 1

我认为您正在使用Tensorflow 1对象检测模型。要使用它们在tensorflow版本 1.15.0 上运行，应该可以使用。这些模型不适用于tensorflow 2，要使用tensorflow 2模型，请使用tensorflow版本2.3.0。