如何適應彩色圖像灰度圖像的算法？

Question

我一直在使用https://gist.github.com/chsasank/4d8f68caf01f041a6453e67fb30f8f5a進行圖像處理。這需要一個只有一個通道的圖像（請參閱assert len(image.shape) == 2）如何調整代碼使其適用於彩色圖像 - 因此具有三個通道的圖像？

Answer 1

最簡單的方法是將圖像分成多個通道，然後在重新映射後合併回來。

# make sure it's not gray 
assert len(image.shape) == 3 

# grab the image resolution 
shape = image.shape[:2] 

# random_state, gaussian and meshgrid 
. 
. 
. 

#calculate indices just as before 
indices = np.reshape(x+dx, (-1, 1)), np.reshape(y+dy, (-1, 1)) 

# split the image into 3 channels 
b, g, r = cv2.split(image) 

# do the mapping on all of the channels separately 
b = map_coordinates(b, indices, order=1).reshape(shape) 
g = map_coordinates(g, indices, order=1).reshape(shape) 
r = map_coordinates(r, indices, order=1).reshape(shape) 

# return merged result 
return cv2.merge((b,g,r))