diff --git a/process/dlc_postprocessing.py b/process/dlc_postprocessing.py index ad01f81d03aa6efde713fc2ba18267b7c3dde47d..451b5fa9575a12646b5feb0d0f2e678471c402f7 100755 --- a/process/dlc_postprocessing.py +++ b/process/dlc_postprocessing.py @@ -717,9 +717,9 @@ def reverse_processes_2d_points(skeleton2d: Dict[str, any], settings: Settings, in_frames = [i for i, f in enumerate(pts_csv['frame']) if f in tmp_set_frames] x_crop = np.interp(skeleton2d[obj_name]['frames'], pts_csv['frame'][in_frames], - pts_csv['x_px'][in_frames], left=np.nan, right=np.nan) - crop_width / 2 + pts_csv['x_px'][in_frames]) - crop_width / 2 y_crop = np.interp(skeleton2d[obj_name]['frames'], pts_csv['frame'][in_frames], - pts_csv['y_px'][in_frames], left=np.nan, right=np.nan) - crop_height / 2 + pts_csv['y_px'][in_frames]) - crop_height / 2 for i in range(0, len(x_crop)): x_crop[i] = np.max([np.min([x_crop[i], width_img - crop_width]), 0])