albumentations-team / albumentations_examples

Augmentations usage examples for albumentations library
MIT License
422 stars 99 forks source link

Add example notebooks with classification and semantic segmentation using PyTorch #8

Closed creafz closed 4 years ago

Dipet commented 4 years ago

I think it is problem with matplotlib.imshow, by default this function works with rgb image, but mask is grayscale image

BloodAxe commented 4 years ago

How about we use ax[...].imshow(mask, interpolation='nearest') on masks to avoid interpolation artifacts when displaying masks then?