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

Posted 2023-02-16

【中文标题】无法使用 Open Vino 工具包将 tensorflow Mask-Rcnn 转换为 IR

【英文标题】：Unable to convert tensorflow Mask-Rcnn to IR with Open Vino toolkit

【发布时间】：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/形状” ]

解决了这个问题。

【参考方案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

时发生异常

与我们分享您的自定义模型以进行复制。您可以将模型保存在云存储中，并在此处向我们提供链接。