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:
@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user