(self, idx)
| 169 | return images, targets |
| 170 | |
| 171 | def __getitem__(self, idx): |
| 172 | sample_start, sample_end, sample_interval = self._get_sample_range(idx) |
| 173 | images, targets = self.pre_continuous_frames(sample_start, sample_end, sample_interval) |
| 174 | data = {} |
| 175 | dataset_name = targets[0]['dataset'] |
| 176 | transform = self.dataset2transform[dataset_name] |
| 177 | if transform is not None: |
| 178 | images, targets = transform(images, targets) |
| 179 | gt_instances = [] |
| 180 | for img_i, targets_i in zip(images, targets): |
| 181 | gt_instances_i = self._targets_to_instances(targets_i, img_i.shape[1:3]) |
| 182 | gt_instances.append(gt_instances_i) |
| 183 | data.update({ |
| 184 | 'imgs': images, |
| 185 | 'gt_instances': gt_instances, |
| 186 | }) |
| 187 | if self.args.vis: |
| 188 | data['ori_img'] = [target_i['ori_img'] for target_i in targets] |
| 189 | return data |
| 190 | |
| 191 | def __len__(self): |
| 192 | return self.item_num |
nothing calls this directly
no test coverage detected