crosstyan da4c51d04f feat: Implement AffinityResult class and optimize camera affinity matrix calculation ...

- Added a new `AffinityResult` class to encapsulate the results of affinity computations, including the affinity matrix, trackings, and their respective indices.
- Introduced a vectorized implementation of `calculate_camera_affinity_matrix_jax` to enhance performance by leveraging JAX's capabilities, replacing the previous double-for-loop approach.
- Updated tests in `test_affinity.py` to include parameterized benchmarks for comparing the performance of the new vectorized method against the naive implementation, ensuring accuracy and efficiency.

2025-04-28 19:08:16 +08:00

6.2 KiB

Raw Blame History

View Raw