這篇博客來源于Stack-Exchange上的一個帖子,問題描述如下:如何將圖中的紅球變為藍球?
這個問題下面有很多答案,我選了最好的一個答案,代碼如下
img = Import["C:/Users/1/Desktop/red.jpg"];
getReds[x_Image] := First@ColorSeparate[x, "Hue"]
isolateSphere[x_Image] := SelectComponents[Binarize[getReds[x], .9], Large]
makeMask[x_Image] := Image@Graphics[Disk @@ (1 /. ComponentMeasurements[isolateSphere[x],{"Centroid", "BoundingDiskRadius"}]),{PlotRange -> Thread[{1, #}], ImageSize -> #} &@ImageDimensions@x]
getAreaToChange[x_Image] := ImageMultiply[x, ColorNegate@makeMask[x]]
shiftColors[x_Image] := Image[ImageData[getAreaToChange[x]] /. p : {r_, g_, b_} /; r > .3 :> RotateLeft[p, 1]]
finishIt[x_Image] := ImageAdd[ImageMultiply[x, makeMask[x]], ColorConvert[shiftColors[x], "RGB"]]
{#, getReds@#, isolateSphere@#, makeMask@#, getAreaToChange@#,shiftColors@#, finishIt@#} &@img // Grid[{#[[1 ;; 2]], #[[3 ;; 5]], #[[6 ;; 7]]}] &
根據結果可以看出,代碼的思路非常明確。分為提取紅色通道分量,生成紅球模板,然后局部轉變為藍色,最后與原圖進行合成。下面是一個
對比結果,
當然,原問題下面還給了很多其他簡短的答案,基本都是上述shiftColors函數的應用。考慮到都是全局處理,存在一定的失真,
就不貼上來了。