无法使用 Open Vino 工具包将 tensorflow Mask-Rcnn 转换为 IR

Question

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” ]

解决了问题。

Answer 1

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"

无法使用 Open Vino 工具包将 tensorflow Mask-Rcnn 转换为 IR

Unable to convert tensorflow Mask-Rcnn to IR with Open Vino toolkit

tensorflow

tensorflow2.0

faster-rcnn

openvino