[workspace] members = [ "ndarray-image", "ndarray-resize", ".", "bounding-box", "ndarray-safetensors", "sqlite3-ndarray-math", "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" } linfa-clustering = { git = "https://github.com/relf/linfa", branch = "upgrade-ndarray-0.16" } [workspace.dependencies] 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", ], branch = "restructure-tensor-type" } mnn-bridge = { git = "https://github.com/uttarayan21/mnn-rs", version = "0.1.0", features = [ "ndarray", ], branch = "restructure-tensor-type" } mnn-sync = { git = "https://github.com/uttarayan21/mnn-rs", version = "0.1.0", features = [ "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" } bytemuck = "1.23.2" error-stack = "0.5.0" thiserror = "2.0" fast_image_resize = "5.2.0" img-parts = "0.4.0" ndarray = { version = "0.16.1", features = ["rayon"] } num = "0.4" [package] name = "detector" version = "0.1.0" edition = "2024" license = "MIT" [dependencies] clap = { version = "4.5", features = ["derive"] } clap_complete = "4.5" error-stack = "0.5" fast_image_resize = "5.2.0" image = "0.25.6" nalgebra = { workspace = true } ndarray = "0.16.1" ndarray-image = { workspace = true } ndarray-resize = { workspace = true } rusqlite = { version = "0.37.0", features = ["functions", "modern-full"] } tap = "1.0.1" thiserror = "2.0" tokio = "1.43.1" tracing = "0.1" tracing-subscriber = { version = "0.3", features = ["env-filter"] } mnn = { workspace = true } mnn-bridge = { workspace = true } mnn-sync = { workspace = true } bounding-box = { version = "0.1.0", path = "bounding-box" } 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", "cuda", ] } ndarray-math = { git = "https://git.darksailor.dev/servius/ndarray-math", version = "0.1.0" } ndarray-safetensors = { version = "0.1.0", path = "ndarray-safetensors" } sqlite3-ndarray-math = { version = "0.1.0", path = "sqlite3-ndarray-math" } # GUI dependencies iced = { version = "0.13", features = ["tokio", "image"] } rfd = "0.15" futures = "0.3" imageproc = "0.25" linfa = "0.7.1" linfa-clustering = "0.7.1" [profile.release] debug = true [features] ort-cuda = [] ort-coreml = ["ort/coreml"] ort-tensorrt = ["ort/tensorrt"] ort-tvm = ["ort/tvm"] ort-openvino = ["ort/openvino"] ort-directml = ["ort/directml"] mnn-metal = ["mnn/metal"] mnn-coreml = ["mnn/coreml"] default = ["ort-cuda"] [[test]] name = "test_bbox_replacement" path = "test_bbox_replacement.rs"