【화상 처리 100개 노크에 도전】Q.9. 가우시안 필터

16006 단어 C++이미지 처리

사용한 라이브러리



【화상 처리 100개 노크】 독자적인 화상 입출력 클래스를 만든다

Q.9. 가우시안 필터



가우시안 필터(3x3, 표준편차 1.3)를 구현하여 imori_noise.jpg의 노이즈를 제거하라.

컨벌루션이군요. 평상시는 적당한 임계치로 잘라, 보다 큰 커널 사이즈로 사용하고 있습니다. 속도가 필요할 때는 FFT를 사용합니다.
int main()
{
    PPM ppm("imori_noise.pnm");
    int width = ppm.Get_width();
    int height = ppm.Get_height();
    PPM ppm2(width, height);

    auto kernel = [&](int i, int j)
    {
        double ret = 2;
        if (i == 0 && j == 0) ret = 4;
        else if (abs(i) == 1 && abs(j)==1) ret = 1;
        return ret;// / 16.;
    };

    auto conv = [&](const std::vector<std::vector<double>> &f)
    {
        std::vector < std::vector < double >> ret(width, std::vector<double>(height));
        for(int j=0; j<height; j++)
            for (int i = 0; i < width; i++)
            {
                ret[i][j] = 0;
            }
        for (int j = 0; j < height; j++)
            for (int i = 0; i < width; i++)
            {
                int sum = 0;
                for(int di=-1; di<=1; di++)
                    for (int dj = -1; dj <= 1; dj++)
                    {
                        if (i - di >= 0 && i - di < width && j - dj >= 0 && j - di < height)
                        {
                            ret[i][j] += kernel(di, dj) * f[i - di][j - dj];
                            sum += kernel(di, dj);
                        }
                    }
                ret[i][j] /= (double)sum;
            }
        return ret;
    };

    std::vector < std::vector < double >> arrr(width, std::vector<double>(height));
    std::vector < std::vector < double >> arrg(width, std::vector<double>(height));
    std::vector < std::vector < double >> arrb(width, std::vector<double>(height));
    for (int j = 0; j < height; j++)
        for (int i = 0; i < width; i++)
        {
            arrr[i][j] = ppm(i, j, 'r');
            arrg[i][j] = ppm(i, j, 'g');
            arrb[i][j] = ppm(i, j, 'b');
        }
    arrr = conv(arrr);
    arrg = conv(arrg);
    arrb = conv(arrb);

    for (int j = 0; j < height; j++)
        for (int i = 0; i < width; i++)
        {
            ppm2(i, j, 'r') = arrr[i][j];
            ppm2(i, j, 'g') = arrg[i][j];
            ppm2(i, j, 'b') = arrb[i][j];
        }

    ppm2.Flush("out.ppm");
    return 0;
}

좋은 웹페이지 즐겨찾기