(self, img_list)
| 37 | return self.transforms(x) |
| 38 | |
| 39 | def forward_batch(self, img_list): |
| 40 | img_batch = [self.transforms(self.to_tensor(img)) for img in img_list] |
| 41 | img_batch = torch.stack(img_batch, dim=0) |
| 42 | return img_batch |
| 43 | |
| 44 | def transform_coords(self, coords: torch.Tensor, normalize=False, orig_hw=None) -> torch.Tensor: |
| 45 | """ |