OpenCV 잡음이 있는 이미지 생성하기 (임펄스 잡음)

잡음이 있는 이미지 생성하기 (임펄스 잡음)

임펄스 잡음 : 흰색과 검은색을 갖는 잡음

c++)

#include <opencv/cv.h>
#include <opencv/highgui.h>
#include <time.h>
 
 
IplImage* ImpulseNoiseCreate(IplImage *img, double noiseRate);
 
 
void main() {
    IplImage* srcImage = cvLoadImage("D:/study/Lena_gray.jpg", -1);
 
 
    IplImage* impulseImage = ImpulseNoiseCreate(srcImage, 0.2);
 
 
    //create window
    cvNamedWindow("impulse", CV_WINDOW_AUTOSIZE);
    //show image
    cvShowImage("impulse", impulseImage);
 
    cvWaitKey(0);
 
    //release image
    cvReleaseImage(&impulseImage);
}
 
//임펄스 잡음 이미지 생성
//임펄스 잡음 : 흰색과 검은색을 갖는 잡음
IplImage* ImpulseNoiseCreate(IplImage *img, double noiseRate) {
    int height = img->height;
    int width = img->width;
    int step = img->widthStep;
    uchar* data = (uchar*)img->imageData;
 
 
    int rand1, rand2;
    int imgSize = width * height;
    int numOfNoise = (int)(imgSize * noiseRate / 2);
    time_t nowTime;
    srand(time(&nowTime));
 
    //랜덤으로 검은색으로 잡음을 준다
    for (int i = 0; i < numOfNoise; i++) {
        rand1 = rand() % width;
        rand2 = rand() % height;
        
        int index = rand1*step + rand2;
        if (index >= imgSize) {
            index = imgSize - 1;
        }
        data[index] = 0;
    }
 
    //랜덤으로 하얀색으로 잡음을 준다
    for (int i = 0; i < numOfNoise; i++) {
        rand1 = rand() % width;
        rand2 = rand() % height;
 
        int index = rand1*step + rand2;
        if (index >= imgSize) {
            index = imgSize - 1;
        }
        data[index] = 255;
    }
 
    return img;
}

결과