机械与电子2024，Vol.42Issue(7)：10-15,6.
基于CycleGan的动车零件图像高光消除方法
A Highlight Elimination Method for Train Parts Image Based on CycleGan
沈钧贤 1梅劲松 1王干 2陈苏扬3
作者信息
- 1. 南京航空航天大学自动化学院,江苏南京 211106
- 2. 南京拓控信息科技股份有限公司,江苏南京 210004
- 3. 南京地铁运营有限责任公司,江苏南京 210000
- 折叠
摘要
Abstract
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.关键词
高光消除/图像处理/光照补偿/循环对抗生成网络Key words
highlight removal/image processing/light compensation/CycleGan分类
信息技术与安全科学引用本文复制引用
沈钧贤,梅劲松,王干,陈苏扬..基于CycleGan的动车零件图像高光消除方法[J].机械与电子,2024,42(7):10-15,6.
