A Highlight Elimination Method for Train Parts Image Based on CycleGan
When using optical non-contact methods to detect damage to motor vehicle parts,the illu-mination light of the collected images can easily cause specular reflection on the parts,resulting in high light on the collected images,which is not conducive to the identification of damaged parts.This paper pro-poses a method based on a combination of CycleGan and illumination compensation algorithm(Retinex)to repair the highlight part of the moving wheel image.First,the highlight area is initially repaired through generation means,then the Retinex illumination compensation algorithm is improved to further enhance the generated image.Experimental results show that the method used greatly improves the pixel grayscale of the highlight area,making the pixel grayscale of this part close to the overall pixel grayscale of the im-age,which achieve the goal of highlight removal.Meanwhile,it strengthens the details and image informa-tion of the dark parts,and improves the recognition of the damaged parts.