search

huggingface / candle

Minimalist ML framework for Rust
Apache License 2.0
15.01k stars 875 forks source link

YoloV9 ONNX: unsupported op_type Identity #1975

Open masc-it opened 5 months ago

masc-it commented 5 months ago

I am trying to run yolov9 exported to onnx (using the export.py script), but I am encountring this error (opset 16):

thread 'main' panicked at src/main.rs:33:72:
called `Result::unwrap()` on an `Err` value: Msg("unsupported op_type Identity for op NodeProto { input: [\"model.49.dfl.conv.weight\"], output: [\"model.49.dfl2.conv.weight\"], name: \"Identity_346\", op_type: \"Identity\", domain: \"\", attribute: [], doc_string: \"\" }")
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Any clues?

jdiaz97 commented 4 months ago

Where did you get the onnx file? I read that the export functionality was a bit broken.

Try these: https://huggingface.co/Xenova/yolov9-onnx/tree/main