Tensorflow 将预训练模型转换为 tflite 时出错 - 不支持 SSD

Question

我从 Tensorflow 2 模型动物园 (ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03) 下载了一个 SSD MobileNet 预训练模型。我的目标是将此预训练模型转换为 tflite。当然，我也许可以找到它已经以 tflite 格式下载，但我需要确保我可以自己转换它们。

我已经安装了 Tensorflow 的 nightly 版本，版本是 '2.5.0-dev20210313' 我还在 https://github.com/tensorflow/models 下载了模型存储库，以确保我拥有所有最新版本的 Python 脚本。

这是我为脚本“export_tflite_graph_tf2.py”运行的命令：

C:\tensorflow2\models\research\object_detection>python export_tflite_graph_tf2.py 
--pipeline_config_path=./ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03/pipeline.config 
--trained_checkpoint_dir=./ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03/ 
--output_directory=./ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03/output

我收到了不支持 SSD MobileNet V2 的错误？这是错误：

<块引用>

2021-03-15 00:44:45.312875：我 张量流/stream_executor/platform/default/dso_loader.cc:49] 成功打开动态库cudart64_110.dll Traceback（最 最近通话最后一次）：文件“export_tflite_graph_tf2.py”，第 161 行，在 app.run(main) File "C:\Users\EnSight\AppData\Roaming\Python\Python37\site-packages\absl\app.py", 第 303 行，运行中 _run_main(main, args) 文件 "C:\Users\EnSight\AppData\Roaming\Python\Python37\site-packages\absl\app.py", 第 251 行，在 _run_main 中 sys.exit(main(argv)) 文件“export_tflite_graph_tf2.py”，第 157 行，在 main FLAGS.centernet_include_keypoints) 文件“C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python37_64\lib\site-packages\object_detection\export_tflite_graph_lib_tf2.py", 第 335 行，在 export_tflite_model 中 pipeline_config.model, is_training=False) 文件“C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python37_64\lib\site-packages\object_detection\builders\model_builder.py", 第 1107 行，构建中 add_summaries) 文件 "C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python37_64\lib\site-packages\object_detection\builders\model_builder.py", 第 377 行，在 _build_ssd_model 中 _check_feature_extractor_exists(ssd_config.feature_extractor.type) 文件“C:\Program Files (x86)\Microsoft Visual Studio\Shared\Python37_64\lib\site-packages\object_detection\builders\model_builder.py", 第 251 行，在 _check_feature_extractor_exists 'Tensorflow'.format(feature_extractor_type)) ValueError：不支持 ssd_mobilenet_v2。请参阅 model_builder.py 了解功能 与不同版本的 Tensorflow 兼容的提取器

我确实在错误中看到我的 Python 版本正在使用不同位置的 object_detection 文件夹，但我更正了这一点并确保它是来自 tensorflow 模型 github 存储库的最新 object_detection 文件夹，因此它应该使用最新的可用的“model_builders.py”版本。

这真的不支持吗？

编辑：我刚刚从头开始设置了一个全新的环境，使用 github 和夜间构建，但我无法让它工作。

谢谢！