forked from HQU-gxy/CVTH3PE
crosstyan
29ca66ad47
- Replaced the `linear_sum_assignment` import from `scipy.optimize` with `hungarian_algorithm` from `optax` to enhance performance in affinity matrix calculations. - Introduced a new `AffinityResult` class to encapsulate results of affinity computations, including trackings and detections, improving the structure of the affinity calculation process. - Removed deprecated functions and debug print statements to streamline the codebase. - Updated `pyproject.toml` and `uv.lock` to include `optax` as a dependency, ensuring compatibility with the new implementation. - Refactored imports and type hints for better organization and consistency across the codebase.
34 lines
747 B
TOML
34 lines
747 B
TOML
[project]
|
|
name = "cvth3pe"
|
|
version = "0.1.0"
|
|
description = "Implementation-of-Cross-View-Tracking-for-Multi-Human-3D-Pose-Estimation-at-over-100-FPS"
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
"anyio>=4.8.0",
|
|
"awkward>=2.7.4",
|
|
"beartype>=0.20.0",
|
|
"cvxopt>=1.3.2",
|
|
"hypothesis>=6.131.9",
|
|
"jax[cuda12]>=0.5.1",
|
|
"jaxtyping>=0.2.38",
|
|
"jupytext>=1.17.0",
|
|
"matplotlib>=3.10.1",
|
|
"opencv-python-headless>=4.11.0.86",
|
|
"optax>=0.2.4",
|
|
"orjson>=3.10.15",
|
|
"pandas>=2.2.3",
|
|
"plotly>=6.0.1",
|
|
"pyarrow>=19.0.1",
|
|
"pytest>=8.3.5",
|
|
"scipy>=1.15.2",
|
|
"torch>=2.6.0",
|
|
"torchvision>=0.21.0",
|
|
"typeguard>=4.4.2",
|
|
]
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"jupyter>=1.1.1",
|
|
]
|