1
0
forked from HQU-gxy/CVTH3PE

feat: Enhance play notebook with new data structures and visualization utilities

- Added new TypedDict classes for camera parameters, including Resolution, Intrinsic, and Extrinsic.
- Updated dataset reading logic to accommodate new camera parameters structure.
- Introduced functions for reading datasets by port and visualizing whole body keypoints.
- Improved the affinity matrix calculation logic in the camera module.
- Updated dependencies in pyproject.toml to include Plotly and SciPy for enhanced functionality.
This commit is contained in:
2025-04-16 18:53:05 +08:00
parent 3f32333de4
commit 3cc93e5eae
8 changed files with 1394 additions and 80 deletions

View File

@ -151,7 +151,7 @@ class _BIPSolver:
clusters[int(label)].append(i)
return list(clusters.values())
def solve(self, affinity_matrix: NDArray, rtn_matrix=False):
def solve(self, affinity_matrix: NDArray):
n_nodes = affinity_matrix.shape[0]
if n_nodes <= 1:
solution_x, sol_matrix = (
@ -197,8 +197,6 @@ class _BIPSolver:
sol_matrix += sol_matrix.T
clusters = self.solution_mat_clusters(sol_matrix)
if not rtn_matrix:
return clusters
return clusters, sol_matrix