(self, idx)
| 233 | return len(self.pairs) |
| 234 | |
| 235 | def __getitem__(self, idx): |
| 236 | name0, name1 = self.pairs[idx] |
| 237 | if self.conf.cache_images: |
| 238 | image0, scale0 = self.images[name0], self.scales[name0] |
| 239 | image1, scale1 = self.images[name1], self.scales[name1] |
| 240 | else: |
| 241 | image0 = read_image(self.image_dir / name0, self.conf.grayscale) |
| 242 | image1 = read_image(self.image_dir / name1, self.conf.grayscale) |
| 243 | image0, scale0 = self.preprocess(image0) |
| 244 | image1, scale1 = self.preprocess(image1) |
| 245 | return image0, image1, scale0, scale1, name0, name1 |
| 246 | |
| 247 | |
| 248 | @torch.no_grad() |
nothing calls this directly
no test coverage detected