[FR] Grayscale to RGB

Hi, I’m using albumentations for my project, but my image is GrayScale with shape [H, W] but I need to use RGB image (3 channels or [H, W, 3]) for training neural networks. Does the library provides any transformation to convert grayscale to RGB, which is similar to tf.image.grayscale_to_rgb? thank you

[domef]

A transformation A.ToRGB would be very usefull.

[hoangtnm]

@BloodAxe I understand your idea, but I think having a transformation like A.RepeatChannels(3) or A.ToRGB will be more easy-to-use rather than use cv2.cvtColor outside A.Compose[...]. Another question is how to scale image in range [0, 255] to [0, 1] like torchvision’s ToTensor, but when I try albumentations.pytorch.transforms.ToTensorV2, it just swaps channels from [H, W, C] to [C, H, W]. Does albumentations have any equivalent transform?