| 178 | return pixel_values, image_grid_thw |
| 179 | |
| 180 | class ImageTransform: |
| 181 | def __init__( |
| 182 | self, |
| 183 | max_image_size, |
| 184 | min_image_size, |
| 185 | image_stride, |
| 186 | max_pixels=14*14*9*1024, |
| 187 | image_mean=[0.5, 0.5, 0.5], |
| 188 | image_std=[0.5, 0.5, 0.5] |
| 189 | ): |
| 190 | self.stride = image_stride |
| 191 | |
| 192 | self.resize_transform = MaxLongEdgeMinShortEdgeResize( |
| 193 | max_size=max_image_size, |
| 194 | min_size=min_image_size, |
| 195 | stride=image_stride, |
| 196 | max_pixels=max_pixels, |
| 197 | ) |
| 198 | self.to_tensor_transform = transforms.ToTensor() |
| 199 | self.normalize_transform = transforms.Normalize(mean=image_mean, std=image_std, inplace=True) |
| 200 | |
| 201 | def __call__(self, img, img_num=1): |
| 202 | img = self.resize_transform(img, img_num=img_num) |
| 203 | img = self.to_tensor_transform(img) |
| 204 | img = self.normalize_transform(img) |
| 205 | return img |
| 206 | |
| 207 | |
| 208 | def decolorization(image): |
no outgoing calls
no test coverage detected