Files
usls/examples/rtdetr
Jamjamjon a0d410b46d Dev (#1)
* Update imageproc crates

* Add top-p method for sampling

* Add SVTR for text recognition & bug fix
2024-04-06 16:16:53 +08:00
..
2024-03-29 15:54:24 +08:00
2024-04-06 16:16:53 +08:00
2024-03-29 15:54:24 +08:00

Quick Start

cargo run -r --example rtdetr

Or you can manully

1. Donwload or export ONNX Model

  • Export

    pip install -U ultralytics
    yolo export model=rtdetr-l.pt format=onnx simplify dynamic opset=16
    
  • Download

    rtdetr-l-f16 model

2. Specify the ONNX model path in main.rs

let options = Options::default()
    .with_model("ONNX_MODEL")    // <= modify this
    .with_saveout("RT-DETR");

3. Then, run

cargo run -r --example rtdetr

Results