History

rUv 004a63e82d fix(security): audit — fix RUSTSEC vulns, clippy warnings, dead code (#769 ) - Upgrade openssl to 0.10.78 (CVE-2026-41676), jsonwebtoken to 9.4 - Suppress unmaintained-only/no-CVE advisories in .cargo/audit.toml with per-entry rationale - Fix all `cargo clippy --all-targets -- -D warnings` errors across 35 crates: derivable_impls, needless_range_loop, map_or→is_some_and/ is_none_or, await_holding_lock (drop MutexGuard before .await), ptr_arg (&mut Vec→&mut [T]), useless_conversion, approximate_constant (2.718→E, 3.14→PI), field_reassign_with_default, manual_inspect, useless_vec, lines_filter_map_ok, print_literal, dead_code - Apply `cargo fmt --all` - Pre-existing test failure in wifi-densepose-signal (test_estimate_occupancy_noise_only) is not introduced by this PR		2026-05-23 05:36:13 -04:00
..
benches	fix(security): audit — fix RUSTSEC vulns, clippy warnings, dead code (#769 )	2026-05-23 05:36:13 -04:00
src	fix(security): audit — fix RUSTSEC vulns, clippy warnings, dead code (#769 )	2026-05-23 05:36:13 -04:00
Cargo.toml	chore(repo): rename rust-port/wifi-densepose-rs → v2/ (flatten to one level) (#427 )	2026-04-25 21:28:13 -04:00
README.md	chore(repo): rename rust-port/wifi-densepose-rs → v2/ (flatten to one level) (#427 )	2026-04-25 21:28:13 -04:00

README.md

wifi-densepose-nn

Multi-backend neural network inference for WiFi-based DensePose estimation.

Overview

wifi-densepose-nn provides the inference engine that maps processed WiFi CSI features to DensePose body surface predictions. It supports three backends -- ONNX Runtime (default), PyTorch via tch-rs, and Candle -- so models can run on CPU, CUDA GPU, or TensorRT depending on the deployment target.

The crate implements two key neural components:

DensePose Head -- Predicts 24 body part segmentation masks and per-part UV coordinate regression.
Modality Translator -- Translates CSI feature embeddings into visual feature space, bridging the domain gap between WiFi signals and image-based pose estimation.

Features

ONNX Runtime backend (default) -- Load and run .onnx models with CPU or GPU execution providers.
PyTorch backend (tch-backend) -- Native PyTorch inference via libtorch FFI.
Candle backend (candle-backend) -- Pure-Rust inference with candle-core and candle-nn.
CUDA acceleration (cuda) -- GPU execution for supported backends.
TensorRT optimization (tensorrt) -- INT8/FP16 optimized inference via ONNX Runtime.
Batched inference -- Process multiple CSI frames in a single forward pass.
Model caching -- Memory-mapped model weights via memmap2.

Feature flags

Flag	Default	Description
`onnx`	yes	ONNX Runtime backend
`tch-backend`	no	PyTorch (tch-rs) backend
`candle-backend`	no	Candle pure-Rust backend
`cuda`	no	CUDA GPU acceleration
`tensorrt`	no	TensorRT via ONNX Runtime
`all-backends`	no	Enable onnx + tch + candle together

Quick Start

use wifi_densepose_nn::{InferenceEngine, DensePoseConfig, OnnxBackend};

// Create inference engine with ONNX backend
let config = DensePoseConfig::default();
let backend = OnnxBackend::from_file("model.onnx")?;
let engine = InferenceEngine::new(backend, config)?;

// Run inference on a CSI feature tensor
let input = ndarray::Array4::zeros((1, 256, 64, 64));
let output = engine.infer(&input)?;

println!("Body parts: {}", output.body_parts.shape()[1]); // 24

Architecture

wifi-densepose-nn/src/
  lib.rs          -- Re-exports, constants (NUM_BODY_PARTS=24), prelude
  densepose.rs    -- DensePoseHead, DensePoseConfig, DensePoseOutput
  inference.rs    -- Backend trait, InferenceEngine, InferenceOptions
  onnx.rs         -- OnnxBackend, OnnxSession (feature-gated)
  tensor.rs       -- Tensor, TensorShape utilities
  translator.rs   -- ModalityTranslator (CSI -> visual space)
  error.rs        -- NnError, NnResult

Crate	Role
`wifi-densepose-core`	Foundation types and `NeuralInference` trait
`wifi-densepose-signal`	Produces CSI features consumed by inference
`wifi-densepose-train`	Trains the models this crate loads
`ort`	ONNX Runtime Rust bindings
`tch`	PyTorch Rust bindings
`candle-core`	Hugging Face pure-Rust ML framework

License

MIT OR Apache-2.0