feat(detector): add CUDA support for ONNX face detection
This commit is contained in:
45
Cargo.toml
45
Cargo.toml
@@ -1,43 +1,42 @@
|
||||
[workspace]
|
||||
members = [
|
||||
"ndarray-image",
|
||||
"ndarray-resize",
|
||||
".",
|
||||
"bounding-box",
|
||||
"ndarray-safetensors",
|
||||
"sqlite3-safetensor-cosine",
|
||||
"ndcv-bridge",
|
||||
"bbox",
|
||||
"ndarray-image",
|
||||
"ndarray-resize",
|
||||
".",
|
||||
"bounding-box",
|
||||
"ndarray-safetensors",
|
||||
"sqlite3-safetensor-cosine",
|
||||
"ndcv-bridge",
|
||||
"bbox",
|
||||
]
|
||||
|
||||
[workspace.package]
|
||||
version = "0.1.0"
|
||||
edition = "2024"
|
||||
|
||||
[patch.crates-io]
|
||||
linfa = { git = "https://github.com/relf/linfa", branch = "upgrade-ndarray-0.16" }
|
||||
|
||||
[workspace.dependencies]
|
||||
bbox = { path = "bbox" }
|
||||
divan = { version = "0.1.21" }
|
||||
ndarray-npy = "0.9.1"
|
||||
serde = { version = "1.0", features = ["derive"] }
|
||||
ndarray-image = { path = "ndarray-image" }
|
||||
ndarray-resize = { path = "ndarray-resize" }
|
||||
mnn = { git = "https://github.com/uttarayan21/mnn-rs", version = "0.2.0", features = [
|
||||
# "metal",
|
||||
# "coreml",
|
||||
"tracing",
|
||||
# "metal",
|
||||
# "coreml",
|
||||
"tracing",
|
||||
], branch = "restructure-tensor-type" }
|
||||
mnn-bridge = { git = "https://github.com/uttarayan21/mnn-rs", version = "0.1.0", features = [
|
||||
"ndarray",
|
||||
"ndarray",
|
||||
], branch = "restructure-tensor-type" }
|
||||
mnn-sync = { git = "https://github.com/uttarayan21/mnn-rs", version = "0.1.0", features = [
|
||||
"tracing",
|
||||
"tracing",
|
||||
], branch = "restructure-tensor-type" }
|
||||
nalgebra = { version = "0.34.0", default-features = false, features = ["std"] }
|
||||
opencv = { version = "0.95.1" }
|
||||
bounding-box = { path = "bounding-box" }
|
||||
ndarray-safetensors = { path = "ndarray-safetensors" }
|
||||
wide = "0.7.33"
|
||||
rayon = "1.11.0"
|
||||
bytemuck = "1.23.2"
|
||||
error-stack = "0.5.0"
|
||||
thiserror = "2.0"
|
||||
@@ -76,9 +75,10 @@ color = "0.3.1"
|
||||
itertools = "0.14.0"
|
||||
ordered-float = "5.0.0"
|
||||
ort = { version = "2.0.0-rc.10", default-features = false, features = [
|
||||
"std",
|
||||
"tracing",
|
||||
"ndarray",
|
||||
"std",
|
||||
"tracing",
|
||||
"ndarray",
|
||||
"cuda",
|
||||
] }
|
||||
ndarray-math = { git = "https://git.darksailor.dev/servius/ndarray-math", version = "0.1.0" }
|
||||
ndarray-safetensors = { version = "0.1.0", path = "ndarray-safetensors" }
|
||||
@@ -89,12 +89,13 @@ iced = { version = "0.13", features = ["tokio", "image"] }
|
||||
rfd = "0.15"
|
||||
futures = "0.3"
|
||||
imageproc = "0.25"
|
||||
linfa = "0.7.1"
|
||||
|
||||
[profile.release]
|
||||
debug = true
|
||||
|
||||
[features]
|
||||
ort-cuda = ["ort/cuda"]
|
||||
ort-cuda = []
|
||||
ort-coreml = ["ort/coreml"]
|
||||
ort-tensorrt = ["ort/tensorrt"]
|
||||
ort-tvm = ["ort/tvm"]
|
||||
@@ -103,7 +104,7 @@ ort-directml = ["ort/directml"]
|
||||
mnn-metal = ["mnn/metal"]
|
||||
mnn-coreml = ["mnn/coreml"]
|
||||
|
||||
default = ["mnn-metal", "mnn-coreml"]
|
||||
default = ["ort-cuda"]
|
||||
|
||||
[[test]]
|
||||
name = "test_bbox_replacement"
|
||||
|
||||
Reference in New Issue
Block a user