Nickyang4900 / Non-extreme-SAP-noise-removal

UESTC 2020 Fall course project<Convex Optimization>

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A Two-stage Method for Non-extreme Value Salt-and-Pepper Noise Removal

Code implementation of UESTC course project: Renwei Yang and YiKe Liu(co-first), 'A Two-stage Method for Non-extreme Value Salt-and-Pepper Noise Removal' [paper]

Abstract

There are several previous methods based on neural network can have great performance in denoising salt and pepper noise. However, those methods work based on a hypothesis that the value of salt and pepper noise is exactly 0 and 255. It is not true in the real world. The result of those method deviation sharply when the value changes form 0 and 255. To overcome this weakness, our method aims at design a convolutional neural network to detect the noise pixels in a wider range of value and then a filter is used to modify it to 0 which is benefit for further work. And another convolutional neural network is used followed the filter to do the denoising and restoration work.

Results

Corrupted, 20% SAP noise at non-extreme value

1

Stage1-Output, noise detection

Blacked-1

Stage2-Output, PSNR = 39.07dB

Output_9

Counterpart, without preprocessing, PSNR = 35.52dB

Counterpart-Output

GT

GT

About

UESTC 2020 Fall course project<Convex Optimization>


Languages

Language:Python 100.0%