Update ort and improve the speed of preprocessing

* Add onnx proto

* Update ort to 2.0.0-rc.2

* Improve the speed of resizing

* Fix yolo-seg bug

* Update README.md
This commit is contained in:
Jamjamjon
2024-05-12 18:12:34 +08:00
committed by GitHub
parent 246e6091cd
commit fc970fc117
48 changed files with 1926 additions and 349 deletions

View File

@ -4,9 +4,9 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
// build model
let options = Options::default()
.with_model("rtdetr-l-f16.onnx")?
.with_confs(&[0.4, 0.15]) // person: 0.4, others: 0.15
.with_confs(&[0.4, 0.15])
.with_names(&coco::NAMES_80);
let mut model = RTDETR::new(&options)?;
let mut model = RTDETR::new(options)?;
// load image
let x = vec![DataLoader::try_read("./assets/bus.jpg")?];