下面的程序爲對Lena圖加入椒鹽噪聲的代碼:
#include <iostream>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/opencv.hpp>
using namespace cv;
using namespace std;
//加入椒鹽噪聲
void salt(Mat image, int n)
{
int i,j;
for (int k=0; k<n; k++)
{
// rand()是隨機數生成器
i= rand()%image.cols;
j= rand()%image.rows;
if (image.type() == CV_8UC1)
{ // 灰度圖像
image.at<uchar>(j,i)= 255;
}
else if (image.type() == CV_8UC3)
{ // 彩色圖像
image.at<cv::Vec3b>(j,i)[0]= 255;
image.at<cv::Vec3b>(j,i)[1]= 255;
image.at<cv::Vec3b>(j,i)[2]= 255;
}
}
}
int main( )
{
Mat image=imread("/Users/zhangxiaoyu/Desktop/lena.jpg");
if(image.empty())
{
cout<<"Error!cannot be read...../n";
return -1;
}
salt(image, 5000);
namedWindow("saltiamge");
imshow("saltimage", image);
waitKey(0);
}
cv::Mat類包含有多種方法,可用來訪問圖像的各種屬性:利用公共成員變量cols和rows可得到圖像的列數和行數;利用cv::Mat的
at(int y,int x)方法可以訪問元素。
結果顯示如下圖所示:
注意調用image.at方法時要注意元素類型:
image.at<uchar>(j,i)=255; (灰度圖像)
彩色圖像的話,每個像素對應了紅,綠,藍,三個通道,因此包含彩色圖像的cv::Mat類會返回一個向量,向量中包含三個8位的數
值。即cv::Vec3b類型,訪問彩色圖像中元素的方法如下:
image.at<cv::Vec3b>(j,i)[channel]=value;