from torchvision.transforms import RandomResizedCrop, ColorJitter, Compose | |
size = ( | |
image_processor.size["shortest_edge"] | |
if "shortest_edge" in image_processor.size | |
else (image_processor.size["height"], image_processor.size["width"]) | |
) | |
_transforms = Compose([RandomResizedCrop(size), ColorJitter(brightness=0.5, hue=0.5)]) | |
The model accepts pixel_values | |
as its input. |