| 59 | |
| 60 | |
| 61 | def preprocess_image(self, image, torch_dtype=None, device=None, pattern="B C H W", min_value=-1, max_value=1): |
| 62 | # Transform a PIL.Image to torch.Tensor |
| 63 | image = torch.Tensor(np.array(image, dtype=np.float32)) |
| 64 | image = image.to(dtype=torch_dtype or self.torch_dtype, device=device or self.device) |
| 65 | image = image * ((max_value - min_value) / 255) + min_value |
| 66 | image = repeat(image, f"H W C -> {pattern}", **({"B": 1} if "B" in pattern else {})) |
| 67 | return image |
| 68 | |
| 69 | |
| 70 | def preprocess_video(self, video, torch_dtype=None, device=None, pattern="B C T H W", min_value=-1, max_value=1): |