Error: DepthToSpace for mode other than CRD is not implemented

[mbkamran]

I am trying to convert an onnx model of Mediapipe into PyTorch for editing it. However, when I try to convert it it gives the following error:

from onnx2torch import convert

person_torch = convert(person_path)
----------------------------------------------------
---------------------------------------------------------------------------
NotImplementedError                       Traceback (most recent call last)
Cell In[2], [line 3](vscode-notebook-cell:?execution_count=2&line=3)
      [1](vscode-notebook-cell:?execution_count=2&line=1) person_path = "../models/onnx/person_detection_mediapipe_2023mar.onnx"
----> [3](vscode-notebook-cell:?execution_count=2&line=3) person_torch = convert(person_path)

File ~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:110, in convert(onnx_model_or_path, save_input_names, attach_onnx_mapping)
    [103](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:103) version = opset_import[onnx_node.domain]
    [104](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:104) converter = get_converter(
    [105](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:105)     domain=onnx_node.domain,
    [106](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:106)     operation_type=onnx_node.operation_type,
    [107](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:107)     version=version,
    [108](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:108) )
--> [110](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:110) torch_module, onnx_mapping = converter(onnx_node, onnx_graph)
    [111](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:111) if attach_onnx_mapping:
    [112](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/converter.py:112)     setattr(torch_module, 'onnx_mapping', onnx_mapping)

File ~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:32, in _(***failed resolving arguments***)
     [29](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:29) mode: str = node.attributes.get('mode', 'DCR')
     [31](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:31) if mode != 'CRD':
---> [32](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:32)     raise NotImplementedError('DepthToSpace for mode other than CRD is not implemented')
     [34](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:34) return OperationConverterResult(
     [35](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:35)     torch_module=OnnxDepthToSpace(blocksize=blocksize),
     [36](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:36)     onnx_mapping=onnx_mapping_from_node(node=node),
     [37](https://file+.vscode-resource.vscode-cdn.net/home/mbkamran/files/mediapipe_pose/onnx2pytorch_conversion/~/anaconda3/envs/env_basic/lib/python3.12/site-packages/onnx2torch/node_converters/depth_to_space.py:37) )

NotImplementedError: DepthToSpace for mode other than CRD is not implemented

Can somebody tell what's the error.