在object_detection_API

问题描述

我正在Tensorflow版本上训练maskrcnn inception v2模型,以便进一步使用OpenVino。训练完模型后,我使用object_detection_API目录中的脚本冻结了模型:

python exporter_main_v2.py \ --trained_checkpoint_dir培训
--output_directory推理图
--pipeline_config_path training / mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8.config

在此脚本之后,我获得了保存的模型和管道文件,以后将在OpenVInO中使用它们 将收到的文件上传到模型优化器时,会发生以下错误:

模型优化器版本: 2020-08-20 11:37:05.425293:我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图之一:

  1. 文本或二进制格式的冻结图
  2. 用于冻结带有文本或二进制格式的检查点(--input_checkpoint)的推理图
  3. 元图

确保为文本的模型提供了--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获取转换后的模型

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)