首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >object_detection_API中重量掩蔽器Tensorflow 2的错误冻结

object_detection_API中重量掩蔽器Tensorflow 2的错误冻结
EN

Stack Overflow用户
提问于 2020-08-20 05:05:25
回答 1查看 597关注 0票数 1

我正在培训Tensorflow版本的maskrcnn初始v2模型,以便与OpenVino进行进一步的工作。在对模型进行培训之后,我使用object_detection_API目录中的脚本冻结了模型:

python exporter_main_v2.py \-训练_检查点_dir训练

--输出目录inference_graph

管道配置路径training/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8.config

在这个脚本之后,我得到了保存的模型和管道文件,这些文件将来应该在OpenVInO中使用--当将接收到的文件上传到模型优化器时会发生以下错误:

模型优化版本: 2020-08-20 11:37:05.425293: i tensorflow/stream_executor/platform/default/dso_loader.cc:44]成功打开动态库cudart64_100.dll框架错误无法加载输入模型: TensorFlow无法读取模型文件:"C:\Users\Anna\Downloads\inference_graph\inference_graph\saved_model\saved_model.pb“是不正确的TensorFlow模型文件。该文件应该包含以下TensorFlow图之一:

文本冻结图或二进制format

  • inference图,用于文本或二进制格式的检查点(--_检查点)冻结
  1. 元图

确保--input_ model _ is _text是为文本格式的模型提供的。默认情况下,模型以二进制格式解释。框架错误详细信息:错误解析消息。有关更多信息,请参考模型优化常见问题(https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_Model_Optimizer_FAQ.html),问题#43。

通过使用我自己的数据集:https://gilberttanner.com/blog/train-a-mask-r-cnn-model-with-the-tensorflow-object-detection-api,我遵循链接文章中的示例来教授模型

在gpu上,模型启动并工作,但我需要得到OpenVINO的转换模型。

EN

回答 1

Stack Overflow用户

发布于 2020-08-28 20:09:15

使用指向mo_tf.py目录的路径运行SavedModel脚本:

python3 _ mo_tf.py -已保存的_

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/63498627

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档