CV

模拟椒盐噪声 salt and pepper

2017-07-16  本文已影响33人  谢小帅

椒盐噪声

源代码

#include <cv.h>
#include <highgui.h>

using namespace cv;
using namespace std;

// n: 噪点个数
// 可以通过image的像素点个数设置一定比例数目的噪点
void salt(Mat image, int n) {
    int i, j;
    for (int k = 0; k < n / 2; ++k) {

        // 随机选取原图的行列
        i = rand() % image.rows;
        j = rand() % image.cols;

        if (image.type() == CV_8UC1) { // 单通道灰度图
            image.at<uchar>(i, j) = 255;
        } else if (image.type() == CV_8UC3) { // color image 彩色图
            image.at<Vec3b>(i, j) = {255, 255, 255};
        }
    }
}

// 椒噪声
// 255 -> 0 即可
void pepper(Mat image, int n) {
    int i, j;
    for (int k = 0; k < n / 2; ++k) {

        // 随机选取原图的行列
        i = rand() % image.rows;
        j = rand() % image.cols;

        if (image.type() == CV_8UC1) { // 单通道灰度图
            image.at<uchar>(i, j) = 0;
        } else if (image.type() == CV_8UC3) { // color image 彩色图
            image.at<Vec3b>(i, j) = {0, 0, 0};
        }
    }
}


int main() {

    Mat src = imread("../pictures/bear.jpeg");

    // 克隆一个一样的Mat,方便后面对同样的数据源添加椒噪声
    // 不能用 Mat src1(src),这样是引用,src与src1的数据一样
    Mat src1 = src.clone();

    // 原图
    namedWindow("src");
    imshow("src", src);

    // 盐噪声
    salt(src, 1000);
    namedWindow("salt");
    imshow("salt", src);

    // 椒噪声
    pepper(src1, 1000);
    namedWindow("pepper");
    imshow("pepper", src1);

    // 椒盐噪声
    pepper(src, 1000);
    namedWindow("salt & pepper");
    imshow("salt & pepper", src);

    waitKey(0);
    return 0;
}

添加噪声的图像


椒盐噪声

可调噪点大小的 pepper 函数

void pepper(Mat image, int n) {
    int i, j;
    for (int k = 0; k < n / 2; ++k) {

        // 随机选取原图的行列
        i = rand() % image.rows;
        j = rand() % image.cols;

        int width = 4;

        if (image.type() == CV_8UC1) { // 单通道灰度图
//            image.at<uchar>(i, j) = 0;
            // 核心代码
            // 4*4 的噪点
            for (int row = 0; row < width; ++row) {
                for (int col = 0; col < width; ++col) {
                    image.at<uchar>((i + row) % image.rows, (j + col) % image.cols) = 0;
                }
            }
        } else if (image.type() == CV_8UC3) { // color image 彩色图
//            image.at<Vec3b>(i, j) = {0, 0, 0};
            for (int row = 0; row < width; ++row) {
                for (int col = 0; col < width; ++col) {
                    image.at<Vec3b>((i + row) % image.rows, (j + col) % image.cols) = {0, 0, 0};
                }
            }
        }
    }
}

放大了的椒噪声


上一篇 下一篇

猜你喜欢

热点阅读