(self, gt_instances: List[Instances])
| 58 | self._current_frame_idx = 0 |
| 59 | |
| 60 | def initialize_for_single_clip(self, gt_instances: List[Instances]): |
| 61 | self.gt_instances = gt_instances |
| 62 | self.num_samples = 0 |
| 63 | self.sample_device = None |
| 64 | self._current_frame_idx = 0 |
| 65 | self.losses_dict = {} |
| 66 | |
| 67 | def _step(self): |
| 68 | self._current_frame_idx += 1 |