Function rgb2bayer

dali/test/python/debayer_test_utils.py:73–79

(img, pattern)

def rgb2bayer(img, pattern):
    h, w, c = img.shape
    assert c == 3
    h = h // 2 * 2
    w = w // 2 * 2
    r, g, b = rgb_bayer_masks((h, w), pattern)
    return img[:h, :w, 0] * r + img[:h, :w, 1] * g + img[:h, :w, 2] * b


def rgb2bayer_seq(seq, patterns):