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

【问题标题】：Unable to convert tensorflow Mask-Rcnn to IR with Open Vino toolkit无法使用 Open Vino 工具包将 tensorflow Mask-Rcnn 转换为 IR
【发布时间】：2022-01-14 03:21:55
【问题描述】：

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”必须在子类中实现。”

我已经使用 exporter_main_v2.py 导出了图表。如果需要更多信息，请告诉我。

编辑：我能够通过更改文件 mask_rcnn_support_api_v2.4.json 来转换模型。

第一个变化： “自定义属性”：{ "operation_to_add": "提案", “clip_before_nms”：假， “clip_after_nms”：真 } 第二个变化： “起点”：[ "StatefulPartitionedCall/concat/concat", "StatefulPartitionedCall/concat_1/concat", "StatefulPartitionedCall/GridAnchorGenerator/Identity", "StatefulPartitionedCall/Cast", "StatefulPartitionedCall/Cast_1", “StatefulPartitionedCall/形状” ]

解决了这个问题。

【问题讨论】：

标签： tensorflow tensorflow2.0 faster-rcnn openvino

【解决方案1】：

OpenVINO 2020.4 与 TensorFlow 2 不兼容。仅在 OpenVINO 2021.3 中完全启用了对 TF 2.0 对象检测 API 模型的支持。

我已使用最新的 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"

【讨论】：

感谢您的回答。我正在使用 OpenVINO 版本 (2021.4.752)。并且仍然出现错误，运行替换器“ObjectDetectionAPIPreprocessor2Replacement () assert len(start_nodes) >= 1 AssertionError 时发生异常
与我们分享您的自定义模型以进行复制。您可以将模型保存在云存储中，并在此处向我们提供链接。