This article describes the pre-processing APIs to convert gray to RGB.

​
Folder Structure

.
├── custom
│ ├── init.py
│ ├── custom_gray2rgb.py
├── train.yaml
└── transforms.yaml

​
Gray to RGB Class

import cv2from avi_transforms import DataItemclass Gray2RGB:
"""Test transform to convert Gray to RGB""" def init(self, defect_map, **kwargs):
# Example to show passing arguments to custom transform
print(defect_map) def call(self, inputs: DataItem) -> DataItem:
image = inputs.image
if image.ndim == 3:
image = image[…, 0]
rgb = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
return DataItem(image=rgb)

​
Use Gray to RGB in transforms.yaml

train:
- GaussianBlur:
lower_limit: 5
upper_limit: 9
p: 0.8
- VerticalFlip:
p: 0.2
- CustomTransform:
transform: custom.custom_gray2rgb.Gray2RGB
params:
defect_map:
0: ok
1: ng
p: 1.0
valid: