mii/usls
1
0
Fork 0
mirror of https://github.com/mii443/usls.git synced 2025-08-22 15:45:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ort

Browse Source
This commit is contained in:
jamjamjon
2024-07-14 10:59:44 +08:00
parent edc3a8897c
commit 56aed9e50c
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Cargo.toml
View File

@ -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" }

2
examples/yolo/main.rs
View File

@ -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),
},