Compared with grayscale images, color images with color information have more advantages in target recognition, detection and tracking, but the color of the image can be easily affected to be distorted by light. Aiming at solving the problem of color distortion, a method based on wavelet transform is proposed to eliminate the influence of illumination. Firstly, the V-channel components in HSV space are decomposed by wavelet and the illumination component in the image scene is estimated by reconstructing the approximate coefficients of wavelet decomposition. Secondly, according to the imaging principle model, the illumination component is removed from the original image, and the reflection component that characterizes the color information is preserved. Finally, the normalized distribution function is used to normalize the reflection components, and the image color recovery is achieved in RGB space. The experimental results show that the proposed method can reduce the influence of non-uniform illumination on the image to restore the color and improve the image quality.