Skip to content

Asymmetric padding is not supported #1370

@AdrianEddy

Description

@AdrianEddy

Describe the bug
I'm trying to run the face detector and landmarker model from MediaPipe.
I first converted the face_detector.tflite to ONNX by running python -m tf2onnx.convert --tflite face_detector.tflite --output face_detector.onnx --opset 15 and I'm trying to run the resulting ONNX models in Burn

When trying to run the ONNX to Burn conversion, I'm getting this log and a panic:

   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   WARN burn_import::onnx::dim_inference: Must implement dimension inference for Pad
   INFO burn_import::onnx::from_onnx: Finished parsing ONNX file: src/model/face_detector.onnx
  DEBUG burn_import::burn::graph: Registering node => 'transpose'
  ERROR burn_import::logger: PANIC => panicked at ~\.cargo\registry\src\index.crates.io-6f17d22bba15001f\burn-import-0.12.1\src\onnx\op_configuration.rs:493:9:
  Asymmetric padding is not supported

build.rs contents:

use burn_import::onnx::ModelGen;
fn main() {
    ModelGen::new()
        .input("src/model/face_detector.onnx")
        .out_dir("model/")
        .run_from_script();
}

To Reproduce
Models I'm trying to run:
face_landmarker.zip

Expected behavior
The conversion succeeds and I can run the model

Desktop (please complete the following information):

  • OS: Windows 11, Rust 1.76

Metadata

Metadata

Assignees

Labels

enhancementEnhance existing features

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions