From 56aed9e50cc53612785a68865e59fd12109fcfd4 Mon Sep 17 00:00:00 2001 From: jamjamjon Date: Sun, 14 Jul 2024 10:59:44 +0800 Subject: [PATCH] Update ort --- Cargo.toml | 4 ++-- examples/yolo/main.rs | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Cargo.toml b/Cargo.toml index 6abece4..93ac94f 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -12,7 +12,7 @@ exclude = ["assets/*", "examples/*"] [dependencies] clap = { version = "4.2.4", features = ["derive"] } ndarray = { version = "0.15.6", features = ["rayon"] } -ort = { version = "2.0.0-rc.2", default-features = false, features = [ +ort = { git = "https://github.com/pykeio/ort.git", default-features = false, features = [ "load-dynamic", "copy-dylibs", "half", @@ -24,7 +24,7 @@ ort = { version = "2.0.0-rc.2", default-features = false, features = [ "rocm", "openvino", "operator-libraries" -] } +], rev = "467d127c5877b099e1d0f605d38b74d221b6121c"} anyhow = { version = "1.0.75" } regex = { version = "1.5.4" } rand = { version = "0.8.5" } diff --git a/examples/yolo/main.rs b/examples/yolo/main.rs index c7838d3..f4889a9 100644 --- a/examples/yolo/main.rs +++ b/examples/yolo/main.rs @@ -116,7 +116,7 @@ fn main() -> Result<()> { }, YOLOVersion::V10 => match args.task { YOLOTask::Detect => { - options.with_model(&args.model.unwrap_or("yolov10n-dyn.onnx".to_string()))? + options.with_model(&args.model.unwrap_or("yolov10n.onnx".to_string()))? } t => anyhow::bail!("Task: {t:?} is unsupported for {:?}", args.version), },