Convert model from ONNX model

General
,
1

Hi, I am trying to convert my ONNX model, but I am encountering an error. My model is not from the model zoo. Most of the components of this model are taken from accelerated_features, but I have customized it a bit.

The model has two inputs:

  1. image with shape (B, 1, H, W)
  2. keypoint with shape (B, N, 2) (where N can be any value greater than 0).

Below is the ONNX model graph:

rkai_model.onnx
rkai_model.onnx1728×8342 306 KB

The error:

image
image1920×1045 127 KB