C++版本OpenCv图像中添加椒盐噪声

C++版本OpenCv教程(二十六)图像中添加椒盐噪声_c++图像加噪声_yhwang-hub的博客-CSDN博客

#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>
using namespace cv;
using namespace std;

//椒盐噪声函数
void saltAndPepper(Mat image, int n) {
    for (int k = 0; k < n; ++k) {
        //随机确定图像中位置
        int i, j;
        i = rand() % image.cols;//取余数运算，保证在图像的列数内
        j = rand() % image.rows;//取余数运算，保证在图像的行数内
        int write_black = rand() % 2;//判定为白色噪声还是黑色噪声的变量
        if (write_black == 0) {//添加白色噪声
            if (image.type() == CV_8UC1) {//处理灰度图像
                image.at<uchar>(j, i) = 255;//白色噪声
            }
            else if (image.type() == CV_8UC3) {//处理彩色图像
                image.at<Vec3b>(j, i)[0] = 255;//Vec3b为opencv定义的3个值的向量类型
                image.at<Vec3b>(j, i)[1] = 255;//[]制定通道，B:0,G:1,R:2
                image.at<Vec3b>(j, i)[2] = 255;
            }
        }
        else {//添加黑噪声
            if (image.type() == CV_8UC1) {
                image.at<uchar>(j, i) = 0;
            }
            else if (image.type() == CV_8UC3) {
                image.at<Vec3b>(j, i)[0] = 0;//Vec3b为opencv定义的3个值的向量类型
                image.at<Vec3b>(j, i)[1] = 0;//[]制定通道，B:0,G:1,R:2
                image.at<Vec3b>(j, i)[2] = 0;
            }
        }
    }
}

int main() {
    Mat img_ = imread("D:/study/opencv/tupian.jpg");
    Mat img;
    resize(img_, img, Size(img_.cols / 2, img_.rows / 2));
    Mat gray;
    cvtColor(img, gray, COLOR_BGR2GRAY);
    if (img.empty() || gray.empty()) {
        cout << "请确认输入的路径是否正确" << endl;
        return -1;
    }
    imshow("luffy原图", img);
    imshow("luffy_gray原图", gray);
    saltAndPepper(img, 10000);//彩色图像添加椒盐噪声
    saltAndPepper(gray, 10000);//灰度图像添加椒盐噪声
    imshow("luffy添加噪声", img);
    imshow("luffy_gray添加噪声", gray);
    waitKey(0);
    return 0;
}