Python中RGB數據到Bayer格式的轉換

Question

我目前正在研究 HIL 模擬。 我正在使用 CARLA 平台，存在一個 RGB 相機傳感器，通過它我能夠獲取圖像幀的 RGB 數據值（3D 數組（高度、寬度、3（RGB）））。 我想通過網絡層發送數據，.....

問題是我想將 RGB 格式轉換為拜耳格式數組。 我不太了解拜耳原始過濾器的細節。 我注意到拜耳到 RGB 的轉換有一個 Opencv 標志，但反之則不然。 我需要這種格式來重用圖像解包或服務器端的庫。

任何關於從 RGB 轉換為 bayer 格式的建議都會幫助我進一步推動這個項目。

參考（圖片參考 - 我想從右到左做）- https://theailearner.com/2018/10/28/bayer-filter/

Answer 1

這里有一個天真的代碼來生成該鏈接的第一Bayer模式您有（GRBG？）：

import numpy as np # of course

im = cv.imread(...)
(height, width) = im.shape[:2]
(B,G,R) = cv.split(im)

bayer = np.empty((height, width), np.uint8)

# strided slicing for this pattern:
#   G R
#   B G
bayer[0::2, 0::2] = G[0::2, 0::2] # top left
bayer[0::2, 1::2] = R[0::2, 1::2] # top right
bayer[1::2, 0::2] = B[1::2, 0::2] # bottom left
bayer[1::2, 1::2] = G[1::2, 1::2] # bottom right

這模擬圖像形成。 您可能希望稍微低通 ( GaussianBlur ) 輸入，否則您會看到有趣的混疊偽影。