As the light travels through the wavefront coding (WFC) system, the modulation transfer function(MTF) of the WFC system was very low, consequently the intermediate blurred image has been received by the detector. However, there is no zero point in the passband of the MTF of the WFC imaging system, and the target information cannot be saved very well. An appropriate filter can be used to restore the sampled intermediate image. The noise of the system is enlarged in the restoration process where the signal be amplified by the filter, and the signal to noise ratio(SNR) of the image is reduced. In order to solve the above issues, an improved algorithm has been proposed in this paper. The noise is controlled by the wavelet in the reconstruction process, and the intermediate blurred image is restored by the wiener filter algorithm with a prior knowledge of the degradation function. Thus, the wavelet de-noising and wiener filter algorithm are combined to restore the middle blurred image of the WFC system. Finally, the restoration image with the diffraction limit level is acquired in image detail restoration and noise control.