| 149 | return img, targets |
| 150 | |
| 151 | def _get_sample_range(self, start_idx): |
| 152 | |
| 153 | # take default sampling method for normal dataset. |
| 154 | assert self.sample_mode in ['fixed_interval', 'random_interval'], 'invalid sample mode: {}'.format(self.sample_mode) |
| 155 | if self.sample_mode == 'fixed_interval': |
| 156 | sample_interval = self.sample_interval |
| 157 | elif self.sample_mode == 'random_interval': |
| 158 | sample_interval = np.random.randint(1, self.sample_interval + 1) |
| 159 | default_range = start_idx, start_idx + (self.num_frames_per_batch - 1) * sample_interval + 1, sample_interval |
| 160 | return default_range |
| 161 | |
| 162 | def pre_continuous_frames(self, start, end, interval=1): |
| 163 | targets = [] |