我在对象检测 API 中使用 ssd_mobilenets 来训练我自己的模型，并获取 .ckpt 文件。它在我的计算机上运行良好，但现在我想在手机上使用该模型。所以，我需要将其转换为 .pb 文件。我不知道该怎么做，有人可以帮忙吗？顺便说一下，ssd_mobilenets 的图很复杂，我找不到哪个是模型的输出。有谁知道输出的名称吗？
Use 导出推理图.py将模型检查点文件转换为 .pb 文件。
python tensorflow_models/object_detection/export_inference_graph.py \
--input_type image_tensor \
--pipeline_config_path architecture_used_while_training.config \
--trained path_to_saved_ckpt/model.ckpt-NUMBER \
--output_directory model/
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
