Python+OpenCV：图像去噪（Image Denoising）

Python+OpenCV：图像去噪（Image Denoising）

理论

We have seen many image smoothing techniques like Gaussian Blurring, Median Blurring etc and they were good to some extent in removing small quantities of noise.

In those techniques, we took a small neighbourhood around a pixel and did some operations like gaussian weighted average, median of the values etc to replace the central element.

In short, noise removal at a pixel was local to its neighbourhood.

There is a property of noise.

Noise is generally considered to be a random variable with zero mean.

Consider a noisy pixel,  where  is the true value of pixel and n is the noise in that pixel.

You can take large number of same pixels (say N) from different images and computes their average.

Ideally, you should get  since mean of noise is zero.

You can verify it yourself by a simple setup.

Hold a static camera to a certain location for a couple of seconds.

This will give you plenty of frames, or a lot of images of the same scene.

Then write a piece of code to find the average of all the frames in the video (This should be too simple for you now ).

Compare the final result and first frame. You can see reduction in noise.

Unfortunately this simple method is not robust to camera and scene motions. Also often there is only one noisy image available.

So idea is simple, we need a set of similar images to average out the noise.

Consider a small window (say 5x5 window) in the image.

Chance is large that the same patch may be somewhere else in the image.

Sometimes in a small neighbourhood around it.

What about using these similar patches together and find their average? For that particular window, that is fine.

See an example image below:

The blue patches in the image looks the similar. Green patches looks similar.

So we take a pixel, take small window around it, search for similar windows in the image, average all the windows and replace the pixel with the result we got.

This method is Non-Local Means Denoising. It takes more time compared to blurring techniques we saw earlier, but its result is very good.

More details and online demo can be found at link: http://www.ipol.im/pub/art/2011/bcm_nlm/ (It has the details, online demo etc. Highly recommended to visit. Our test image is generated from this link).

For color images, image is converted to CIELAB colorspace and then it separately denoise L and AB components.

Image Denoising in OpenCV

OpenCV provides four variations of this technique.

1. cv.fastNlMeansDenoising() - works with a single grayscale images.
2. cv.fastNlMeansDenoisingColored() - works with a color image.
3. cv.fastNlMeansDenoisingMulti() - works with image sequence captured in short period of time (grayscale images).
4. cv.fastNlMeansDenoisingColoredMulti() - same as above, but for color images.

Common arguments are:

• h : parameter deciding filter strength. Higher h value removes noise better, but removes details of image also. (10 is ok).
• hForColorComponents : same as h, but for color images only. (normally same as h).
• templateWindowSize : should be odd. (recommended 7).
• searchWindowSize : should be odd. (recommended 21).

Please visit first link in additional resources for more details on these parameters.

We will demonstrate 2 and 3 here. Rest is left for you.

1. cv.fastNlMeansDenoisingColored()

As mentioned above it is used to remove noise from color images. (Noise is expected to be gaussian).

See the example below:

####################################################################################################
# 图像去噪(Image Denoising)
def lmc_cv_image_denoising(method):
    """
        函数功能: 图像去噪(Image Denoising).
    """

    #  0: fastNlMeansDenoisingColored(): used to remove noise from color images.
    if 0 == method:
        image = lmc_cv.imread('D:/99-Research/TestData/image/cartoon6.jpg')
        image = lmc_cv.cvtColor(image, lmc_cv.COLOR_BGR2RGB)
        denoising_image = lmc_cv.fastNlMeansDenoisingColored(image, None, 10, 10, 7, 21)
        # 显示结果
        pyplot.subplot(121)
        pyplot.imshow(image)
        pyplot.xticks([])
        pyplot.yticks([])
        pyplot.subplot(122)
        pyplot.imshow(denoising_image)
        pyplot.xticks([])
        pyplot.yticks([])
        pyplot.show()

2. cv.fastNlMeansDenoisingMulti()

Now we will apply the same method to a video.

The first argument is the list of noisy frames.

Second argument imgToDenoiseIndex specifies which frame we need to denoise, for that we pass the index of frame in our input list.

Third is the temporalWindowSize which specifies the number of nearby frames to be used for denoising.

It should be odd. In that case, a total of temporalWindowSize frames are used where central frame is the frame to be denoised.

For example, you passed a list of 5 frames as input. Let imgToDenoiseIndex = 2 and temporalWindowSize = 3. Then frame-1, frame-2 and frame-3 are used to denoise frame-2.

Let's see an example.

####################################################################################################
# 图像去噪(Image Denoising)
def lmc_cv_image_denoising(method):
    """
        函数功能: 图像去噪(Image Denoising).
    """


    # 1: remove noise from color video.
    if 1 == method:
        # 读取视频
        parser = argparse.ArgumentParser(
            description='This sample demonstrates remove noise from color video.')
        parser.add_argument('--input', type=str, help='Path to a video or a sequence of image.',
                            default='D:/99-Research/TestData/image/slow_traffic_small.mp4')
        args = parser.parse_args()
        cap = lmc_cv.VideoCapture(lmc_cv.samples.findFileOrKeep(args.input))

        # create a list of first 5 frames
        image = [cap.read()[1] for i in range(5)]

        # convert all to grayscale
        gray = [lmc_cv.cvtColor(i, lmc_cv.COLOR_BGR2GRAY) for i in image]

        # convert all to float64
        gray = [np.float64(i) for i in gray]

        # create a noise of variance 25
        noise = np.random.randn(*gray[1].shape) * 10

        # Add this noise to images
        noisy = [i + noise for i in gray]

        # Convert back to uint8
        noisy = [np.uint8(np.clip(i, 0, 255)) for i in noisy]

        # Denoise 3rd frame considering all the 5 frames
        denoising_image = lmc_cv.fastNlMeansDenoisingMulti(noisy, 2, 5, None, 4, 7, 35)

        # 显示结果
        pyplot.subplot(131)
        pyplot.imshow(gray[2], 'gray')
        pyplot.xticks([])
        pyplot.yticks([])
        pyplot.subplot(132)
        pyplot.imshow(noisy[2], 'gray')
        pyplot.xticks([])
        pyplot.yticks([])
        pyplot.subplot(133)
        pyplot.imshow(denoising_image, 'gray')
        pyplot.xticks([])
        pyplot.yticks([])
        pyplot.show()

It takes considerable amount of time for computation. In the result, first image is the original frame, second is the noisy one, third is the denoised image.