search

jbohnslav / opencv_transforms

OpenCV implementation of Torchvision's image augmentations
MIT License
375 stars 46 forks source link

Normalize giving an error #24

Open shreygupta2809 opened 1 year ago

shreygupta2809 commented 1 year ago 

RESIZE = (128, 171)
MEAN = [0.43216, 0.394666, 0.37645]
STD = [0.22803, 0.22145, 0.216989]
CROP_SIZE = 112

transform = Compose(
                [
                    # Resize(RESIZE),
                    # CenterCrop(CROP_SIZE),
                    # ToTensor(),
                    Normalize(MEAN, STD)
                ]
            )

On trying the above code (even without commenting) on google collab I get the following error: image Does anyone know what the issue is or how to fix this? Attaching list of pip packages in colab for reference if needed packages.txt

Thanks