(
self,
motion_file: str,
input_fps: int,
output_fps: int,
device: torch.device,
frame_range: tuple[int, int] | None,
)
| 77 | |
| 78 | class MotionLoader: |
| 79 | def __init__( |
| 80 | self, |
| 81 | motion_file: str, |
| 82 | input_fps: int, |
| 83 | output_fps: int, |
| 84 | device: torch.device, |
| 85 | frame_range: tuple[int, int] | None, |
| 86 | ): |
| 87 | self.motion_file = motion_file |
| 88 | self.input_fps = input_fps |
| 89 | self.output_fps = output_fps |
| 90 | self.input_dt = 1.0 / self.input_fps |
| 91 | self.output_dt = 1.0 / self.output_fps |
| 92 | self.current_idx = 0 |
| 93 | self.device = device |
| 94 | self.frame_range = frame_range |
| 95 | self._load_motion() |
| 96 | self._interpolate_motion() |
| 97 | self._compute_velocities() |
| 98 | |
| 99 | def _load_motion(self): |
| 100 | """Loads the motion from the csv file.""" |
nothing calls this directly
no test coverage detected