feat: Complete Rust port of WiFi-DensePose with modular crates
Major changes: - Organized Python v1 implementation into v1/ subdirectory - Created Rust workspace with 9 modular crates: - wifi-densepose-core: Core types, traits, errors - wifi-densepose-signal: CSI processing, phase sanitization, FFT - wifi-densepose-nn: Neural network inference (ONNX/Candle/tch) - wifi-densepose-api: Axum-based REST/WebSocket API - wifi-densepose-db: SQLx database layer - wifi-densepose-config: Configuration management - wifi-densepose-hardware: Hardware abstraction - wifi-densepose-wasm: WebAssembly bindings - wifi-densepose-cli: Command-line interface Documentation: - ADR-001: Workspace structure - ADR-002: Signal processing library selection - ADR-003: Neural network inference strategy - DDD domain model with bounded contexts Testing: - 69 tests passing across all crates - Signal processing: 45 tests - Neural networks: 21 tests - Core: 3 doc tests Performance targets: - 10x faster CSI processing (~0.5ms vs ~5ms) - 5x lower memory usage (~100MB vs ~500MB) - WASM support for browser deployment
This commit is contained in:
Claude
218
v1/setup.py
Normal file
218
v1/setup.py
Normal file
@@ -0,0 +1,218 @@
|
||||
"""
|
||||
Setup script for WiFi-DensePose API
|
||||
This file is maintained for backward compatibility.
|
||||
The main configuration is in pyproject.toml.
|
||||
"""
|
||||
|
||||
from setuptools import setup, find_packages
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# Ensure we're in the right directory
|
||||
if __name__ == "__main__":
|
||||
here = Path(__file__).parent.absolute()
|
||||
os.chdir(here)
|
||||
|
||||
# Read version from src/__init__.py
|
||||
def get_version():
|
||||
"""Get version from src/__init__.py"""
|
||||
version_file = here / "src" / "__init__.py"
|
||||
if version_file.exists():
|
||||
with open(version_file, 'r') as f:
|
||||
for line in f:
|
||||
if line.startswith('__version__'):
|
||||
return line.split('=')[1].strip().strip('"').strip("'")
|
||||
return "1.0.0"
|
||||
|
||||
# Read long description from README
|
||||
def get_long_description():
|
||||
"""Get long description from README.md"""
|
||||
readme_file = here / "README.md"
|
||||
if readme_file.exists():
|
||||
with open(readme_file, 'r', encoding='utf-8') as f:
|
||||
return f.read()
|
||||
return "WiFi-based human pose estimation using CSI data and DensePose neural networks"
|
||||
|
||||
# Read requirements from requirements.txt if it exists
|
||||
def get_requirements():
|
||||
"""Get requirements from requirements.txt or use defaults"""
|
||||
requirements_file = here / "requirements.txt"
|
||||
if requirements_file.exists():
|
||||
with open(requirements_file, 'r') as f:
|
||||
return [line.strip() for line in f if line.strip() and not line.startswith('#')]
|
||||
|
||||
# Default requirements (should match pyproject.toml)
|
||||
return [
|
||||
"fastapi>=0.104.0",
|
||||
"uvicorn[standard]>=0.24.0",
|
||||
"pydantic>=2.5.0",
|
||||
"pydantic-settings>=2.1.0",
|
||||
"sqlalchemy>=2.0.0",
|
||||
"alembic>=1.13.0",
|
||||
"asyncpg>=0.29.0",
|
||||
"psycopg2-binary>=2.9.0",
|
||||
"redis>=5.0.0",
|
||||
"aioredis>=2.0.0",
|
||||
"torch>=2.1.0",
|
||||
"torchvision>=0.16.0",
|
||||
"numpy>=1.24.0",
|
||||
"opencv-python>=4.8.0",
|
||||
"pillow>=10.0.0",
|
||||
"scikit-learn>=1.3.0",
|
||||
"scipy>=1.11.0",
|
||||
"matplotlib>=3.7.0",
|
||||
"pandas>=2.1.0",
|
||||
"scapy>=2.5.0",
|
||||
"pyserial>=3.5",
|
||||
"paramiko>=3.3.0",
|
||||
"click>=8.1.0",
|
||||
"rich>=13.6.0",
|
||||
"typer>=0.9.0",
|
||||
"python-multipart>=0.0.6",
|
||||
"python-jose[cryptography]>=3.3.0",
|
||||
"passlib[bcrypt]>=1.7.4",
|
||||
"python-dotenv>=1.0.0",
|
||||
"pyyaml>=6.0",
|
||||
"toml>=0.10.2",
|
||||
"prometheus-client>=0.19.0",
|
||||
"structlog>=23.2.0",
|
||||
"psutil>=5.9.0",
|
||||
"httpx>=0.25.0",
|
||||
"aiofiles>=23.2.0",
|
||||
"marshmallow>=3.20.0",
|
||||
"jsonschema>=4.19.0",
|
||||
"celery>=5.3.0",
|
||||
"kombu>=5.3.0",
|
||||
]
|
||||
|
||||
# Development requirements
|
||||
def get_dev_requirements():
|
||||
"""Get development requirements"""
|
||||
return [
|
||||
"pytest>=7.4.0",
|
||||
"pytest-asyncio>=0.21.0",
|
||||
"pytest-cov>=4.1.0",
|
||||
"pytest-mock>=3.12.0",
|
||||
"pytest-xdist>=3.3.0",
|
||||
"black>=23.9.0",
|
||||
"isort>=5.12.0",
|
||||
"flake8>=6.1.0",
|
||||
"mypy>=1.6.0",
|
||||
"pre-commit>=3.5.0",
|
||||
"bandit>=1.7.0",
|
||||
"safety>=2.3.0",
|
||||
]
|
||||
|
||||
# Check Python version
|
||||
if sys.version_info < (3, 9):
|
||||
sys.exit("Python 3.9 or higher is required")
|
||||
|
||||
# Setup configuration
|
||||
setup(
|
||||
name="wifi-densepose",
|
||||
version=get_version(),
|
||||
description="WiFi-based human pose estimation using CSI data and DensePose neural networks",
|
||||
long_description=get_long_description(),
|
||||
long_description_content_type="text/markdown",
|
||||
|
||||
# Author information
|
||||
author="rUv",
|
||||
author_email="ruv@ruv.net",
|
||||
maintainer="rUv",
|
||||
maintainer_email="ruv@ruv.net",
|
||||
|
||||
# URLs
|
||||
url="https://github.com/ruvnet/wifi-densepose",
|
||||
project_urls={
|
||||
"Documentation": "https://github.com/ruvnet/wifi-densepose#readme",
|
||||
"Source": "https://github.com/ruvnet/wifi-densepose",
|
||||
"Tracker": "https://github.com/ruvnet/wifi-densepose/issues",
|
||||
},
|
||||
|
||||
# Package configuration
|
||||
packages=find_packages(include=["src", "src.*"]),
|
||||
package_dir={"": "."},
|
||||
|
||||
# Include package data
|
||||
package_data={
|
||||
"src": [
|
||||
"*.yaml", "*.yml", "*.json", "*.toml", "*.cfg", "*.ini"
|
||||
],
|
||||
"src.models": ["*.pth", "*.onnx", "*.pt"],
|
||||
"src.config": ["*.yaml", "*.yml", "*.json"],
|
||||
},
|
||||
include_package_data=True,
|
||||
|
||||
# Requirements
|
||||
python_requires=">=3.9",
|
||||
install_requires=get_requirements(),
|
||||
extras_require={
|
||||
"dev": get_dev_requirements(),
|
||||
"docs": [
|
||||
"sphinx>=7.2.0",
|
||||
"sphinx-rtd-theme>=1.3.0",
|
||||
"sphinx-autodoc-typehints>=1.25.0",
|
||||
"myst-parser>=2.0.0",
|
||||
],
|
||||
"gpu": [
|
||||
"torch>=2.1.0",
|
||||
"torchvision>=0.16.0",
|
||||
"nvidia-ml-py>=12.535.0",
|
||||
],
|
||||
"monitoring": [
|
||||
"grafana-api>=1.0.3",
|
||||
"influxdb-client>=1.38.0",
|
||||
"elasticsearch>=8.10.0",
|
||||
],
|
||||
"deployment": [
|
||||
"gunicorn>=21.2.0",
|
||||
"docker>=6.1.0",
|
||||
"kubernetes>=28.1.0",
|
||||
],
|
||||
},
|
||||
|
||||
# Entry points
|
||||
entry_points={
|
||||
"console_scripts": [
|
||||
"wifi-densepose=src.cli:cli",
|
||||
"wdp=src.cli:cli",
|
||||
],
|
||||
"wifi_densepose.plugins": [
|
||||
# Plugin entry points for extensibility
|
||||
],
|
||||
},
|
||||
|
||||
# Classification
|
||||
classifiers=[
|
||||
"Development Status :: 4 - Beta",
|
||||
"Intended Audience :: Developers",
|
||||
"Intended Audience :: Science/Research",
|
||||
"License :: OSI Approved :: MIT License",
|
||||
"Operating System :: OS Independent",
|
||||
"Programming Language :: Python :: 3",
|
||||
"Programming Language :: Python :: 3.9",
|
||||
"Programming Language :: Python :: 3.10",
|
||||
"Programming Language :: Python :: 3.11",
|
||||
"Programming Language :: Python :: 3.12",
|
||||
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
||||
"Topic :: Scientific/Engineering :: Image Processing",
|
||||
"Topic :: System :: Networking",
|
||||
"Topic :: Software Development :: Libraries :: Python Modules",
|
||||
],
|
||||
|
||||
# Keywords
|
||||
keywords=[
|
||||
"wifi", "csi", "pose-estimation", "densepose", "neural-networks",
|
||||
"computer-vision", "machine-learning", "iot", "wireless-sensing"
|
||||
],
|
||||
|
||||
# License
|
||||
license="MIT",
|
||||
|
||||
# Zip safe
|
||||
zip_safe=False,
|
||||
|
||||
# Platform
|
||||
platforms=["any"],
|
||||
)
|
||||
Reference in New Issue
Block a user