我正在培训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
确保--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的转换模型。
发布于 2020-08-28 20:09:15
使用指向mo_tf.py目录的路径运行SavedModel脚本:
python3 _ mo_tf.py -已保存的_
https://stackoverflow.com/questions/63498627
复制相似问题