无法使用 Open Vino 工具包将 tensorflow Mask-Rcnn 转换为 IR
Unable to convert tensorflow Mask-Rcnn to IR with Open Vino toolkit
python mo_tf.py
--saved_model_dir C:\DATASETS\mask50000\exports\saved_model
--output_dir C:\DATASETS\mask50000
--reverse_input_channels
--tensorflow_custom_operations_config extensions\front\tf\mask_rcnn_support_api_v2.0.json
--tensorflow_object_detection_api_pipeline_config C:\DATASETS\mask50000\exports\pipeline.config
--log_level=DEBUG
我一直在尝试使用上面的脚本转换模型,但每次都出现错误:
“异常:在 运行 替换器“REPLACEMENT_ID () 期间发生异常:函数 'update_custom_layer_attributes' 必须是在子class.
中实现
我已经使用 exporter_main_v2.py 导出了图表。如果需要更多信息,请通知我。
编辑:
我能够通过更改文件 mask_rcnn_support_api_v2.4.json.
来转换模型
第一个变化:
“custom_attributes”:{
"operation_to_add": "提案",
“clip_before_nms”:错误,
“clip_after_nms”:真
}
第二个变化:
“start_points”:[
"StatefulPartitionedCall/concat/concat",
"StatefulPartitionedCall/concat_1/concat",
"StatefulPartitionedCall/GridAnchorGenerator/Identity",
"StatefulPartitionedCall/Cast",
"StatefulPartitionedCall/Cast_1",
“StatefulPartitionedCall/Shape”
]
解决了问题。
OpenVINO 2020.4 与 TensorFlow 2 不兼容。支持 TF 2.0 对象检测 API 模型仅在 OpenVINO 2021.3 中完全启用。
我已经使用最新的 OpenVINO 版本 (2021.4.752) 成功地将模型 mask_rcnn_inception_resnet_v2_1024x1024_coco17 转换为 IR。
我在这里分享MO转换命令:
python mo_tf.py --saved_model_dir \saved_model --tensorflow_object_detection_api_pipeline_config \pipeline.config --transformations_config \extensions\front\tf\mask_rcnn_support_api_v2.0.json"
python mo_tf.py
--saved_model_dir C:\DATASETS\mask50000\exports\saved_model
--output_dir C:\DATASETS\mask50000
--reverse_input_channels
--tensorflow_custom_operations_config extensions\front\tf\mask_rcnn_support_api_v2.0.json
--tensorflow_object_detection_api_pipeline_config C:\DATASETS\mask50000\exports\pipeline.config
--log_level=DEBUG
我一直在尝试使用上面的脚本转换模型,但每次都出现错误:
“异常:在 运行 替换器“REPLACEMENT_ID (
我已经使用 exporter_main_v2.py 导出了图表。如果需要更多信息,请通知我。
编辑: 我能够通过更改文件 mask_rcnn_support_api_v2.4.json.
来转换模型第一个变化: “custom_attributes”:{ "operation_to_add": "提案", “clip_before_nms”:错误, “clip_after_nms”:真 } 第二个变化: “start_points”:[ "StatefulPartitionedCall/concat/concat", "StatefulPartitionedCall/concat_1/concat", "StatefulPartitionedCall/GridAnchorGenerator/Identity", "StatefulPartitionedCall/Cast", "StatefulPartitionedCall/Cast_1", “StatefulPartitionedCall/Shape” ]
解决了问题。
OpenVINO 2020.4 与 TensorFlow 2 不兼容。支持 TF 2.0 对象检测 API 模型仅在 OpenVINO 2021.3 中完全启用。
我已经使用最新的 OpenVINO 版本 (2021.4.752) 成功地将模型 mask_rcnn_inception_resnet_v2_1024x1024_coco17 转换为 IR。
我在这里分享MO转换命令:
python mo_tf.py --saved_model_dir