convert raw dataset images to class images

[SharhadBashar]

Hi, all the images in the drop box are in raw format: Is there a way to add class level colouring to them, so they look like this?

[DrSleep]

take a look at this function