在用OpenCV對圖像進行處理時,利用顏色定位是常常會接觸到的方法,但RGB受光照影響比較嚴重,轉換到HSV XYZ等空間也解決不了時,
可以用白平衡算法進行修正,使其發黃、發藍、發紅的照片更加趨於自然光下的圖像。(轉摘請說明來源)
程序代碼示例如下:
1 //該代碼實現白平衡算法中的灰度世界法,能有效改善圖像發紅發藍發綠的現象; 2 3 #include <opencv2/opencv.hpp> 4 using namespace cv; 5 6 int main() 7 { 8 Mat g_srcImage,dstImage; 9 vector<Mat> g_vChannels; 10 g_srcImage = imread("C:/Users/Administrator/Desktop/區分高架定位/01.jpg"); 11 imshow("原圖",g_srcImage); 12 //waitKey(0); 13 14 //分離通道 15 split(g_srcImage,g_vChannels); 16 Mat imageBlueChannel = g_vChannels.at(0); 17 Mat imageGreenChannel = g_vChannels.at(1); 18 Mat imageRedChannel = g_vChannels.at(2); 19 20 double imageBlueChannelAvg=0; 21 double imageGreenChannelAvg=0; 22 double imageRedChannelAvg=0; 23 24 //求各通道的平均值 25 imageBlueChannelAvg = mean(imageBlueChannel)[0]; 26 imageGreenChannelAvg = mean(imageGreenChannel)[0]; 27 imageRedChannelAvg = mean(imageRedChannel)[0]; 28 29 //求出個通道所占增益 30 double K = (imageRedChannelAvg+imageGreenChannelAvg+imageRedChannelAvg)/3; 31 double Kb = K/imageBlueChannelAvg; 32 double Kg = K/imageGreenChannelAvg; 33 double Kr = K/imageRedChannelAvg; 34 35 //更新白平衡后的各通道BGR值 36 addWeighted(imageBlueChannel,Kb,0,0,0,imageBlueChannel); 37 addWeighted(imageGreenChannel,Kg,0,0,0,imageGreenChannel); 38 addWeighted(imageRedChannel,Kr,0,0,0,imageRedChannel); 39 40 merge(g_vChannels,dstImage);//圖像各通道合並 41 imshow("白平衡后圖",dstImage); 42 waitKey(0); 43 return 0; 44 }
結果如下: