MCPcopy Create free account
hub / github.com/DeepLabCut/DeepLabCut / _undistort_views

Function _undistort_views

deeplabcut/pose_estimation_3d/triangulation.py:485–501  ·  view source on GitHub ↗
(df_view_pairs, stereo_params)

Source from the content-addressed store, hash-verified

483
484
485def _undistort_views(df_view_pairs, stereo_params):
486 df_views_undist = []
487 for df_view_pair, camera_pair in zip(df_view_pairs, stereo_params, strict=False):
488 params = stereo_params[camera_pair]
489 dfs = []
490 for i, df_view in enumerate(df_view_pair, start=1):
491 pts_undist = _undistort_points(
492 df_view.to_numpy(),
493 params[f"cameraMatrix{i}"],
494 params[f"distCoeffs{i}"],
495 params[f"P{i}"],
496 params[f"R{i}"],
497 )
498 df = pd.DataFrame(pts_undist, df_view.index, df_view.columns)
499 dfs.append(df)
500 df_views_undist.append(dfs)
501 return df_views_undist
502
503
504def undistort_points(config, dataframe, camera_pair):

Callers 1

undistort_pointsFunction · 0.85

Calls 1

_undistort_pointsFunction · 0.85

Tested by

no test coverage detected