-
Notifications
You must be signed in to change notification settings - Fork 769
Open
Labels
enhancementEnhance existing featuresEnhance existing features
Description
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
jackrr, antimora and raphaelmenges
Metadata
Metadata
Assignees
Labels
enhancementEnhance existing featuresEnhance existing features