search

jbohnslav / opencv_transforms

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

Fix the gray image checking error #13

Open Larry-u opened 4 years ago

Larry-u commented 4 years ago

The default dimension of a gray image read by opencv is 2, which means using img.shape[2] will raise index error.