flystem
/
flystem-usls
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update ort

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
Show Stats 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] [dependencies]
clap = { version = "4.2.4", features = ["derive"] } clap = { version = "4.2.4", features = ["derive"] }
ndarray = { version = "0.15.6", features = ["rayon"] } 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", "load-dynamic",
"copy-dylibs", "copy-dylibs",
"half", "half",
@ -24,7 +24,7 @@ ort = { version = "2.0.0-rc.2", default-features = false, features = [
"rocm", "rocm",
"openvino", "openvino",
"operator-libraries" "operator-libraries"
] } ], rev = "467d127c5877b099e1d0f605d38b74d221b6121c"}
anyhow = { version = "1.0.75" } anyhow = { version = "1.0.75" }
regex = { version = "1.5.4" } regex = { version = "1.5.4" }
rand = { version = "0.8.5" } rand = { version = "0.8.5" }

2
examples/yolo/main.rs
View File

@ -116,7 +116,7 @@ fn main() -> Result<()> {
}, },
YOLOVersion::V10 => match args.task { YOLOVersion::V10 => match args.task {
YOLOTask::Detect => { 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), t => anyhow::bail!("Task: {t:?} is unsupported for {:?}", args.version),
}, },