Pre Processing

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)

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: