servius/face-detector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(detector): add CUDA support for ONNX face detection
Some checks failed
build / checks-build (push) Has been cancelled
Details
build / codecov (push) Has been cancelled
Details
docs / docs (push) Has been cancelled
Details
build / checks-matrix (push) Has been cancelled
Details

Browse Source
This commit is contained in:
uttarayan21
2025-08-28 18:32:00 +05:30
parent 4256c0af74
commit ac8f1d01b4
10 changed files with 315 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
Cargo.lock generated
View File

@@ -1271,13 +1271,14 @@ dependencies = [
"image 0.25.6",
"imageproc",
"itertools 0.14.0",
"linfa",
"mnn",
"mnn-bridge",
"mnn-sync",
"nalgebra 0.34.0",
"ndarray",
"ndarray-image",
"ndarray-math 0.1.0 (git+https://git.darksailor.dev/servius/ndarray-math)",
"ndarray-math",
"ndarray-resize",
"ndarray-safetensors",
"ordered-float",
@@ -2916,6 +2917,19 @@ dependencies = [
"vcpkg",
]
[[package]]
name = "linfa"
version = "0.7.1"
source = "git+https://github.com/relf/linfa?branch=upgrade-ndarray-0.16#c1fbee7c54e806de3f5fb2c5240ce163d000f1ba"
dependencies = [
"approx",
"ndarray",
"num-traits",
"rand 0.8.5",
"sprs",
"thiserror 2.0.15",
]
[[package]]
name = "linux-raw-sys"
version = "0.4.15"
@@ -3223,6 +3237,7 @@ version = "0.16.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "882ed72dce9365842bf196bdeedf5055305f11fc8c03dee7bb0194a6cad34841"
dependencies = [
"approx",
"matrixmultiply",
"num-complex",
"num-integer",
@@ -3244,16 +3259,7 @@ dependencies = [
[[package]]
name = "ndarray-math"
version = "0.1.0"
dependencies = [
"ndarray",
"num",
"thiserror 2.0.15",
]
[[package]]
name = "ndarray-math"
version = "0.1.0"
source = "git+https://git.darksailor.dev/servius/ndarray-math#f047966f20835267f20e5839272b9ab36c445796"
source = "git+https://git.darksailor.dev/servius/ndarray-math#df17c36193df60e070e4e120c9feebe68ff3f517"
dependencies = [
"ndarray",
"num",
@@ -5120,6 +5126,18 @@ dependencies = [
"bitflags 2.9.2",
]
[[package]]
name = "sprs"
version = "0.11.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8bff8419009a08f6cb7519a602c5590241fbff1446bcc823c07af15386eb801b"
dependencies = [
"ndarray",
"num-complex",
"num-traits",
"smallvec 1.15.1",
]
[[package]]
name = "sqlite-loadable"
version = "0.0.5"
@@ -5149,7 +5167,7 @@ name = "sqlite3-safetensor-cosine"
version = "0.1.0"
dependencies = [
"ndarray",
"ndarray-math 0.1.0",
"ndarray-math",
"ndarray-safetensors",
"sqlite-loadable",
]