A pupil detection method based on Unet with attention module and shape-prior loss
文献类型:会议论文
作者 | Song, Wenhui1,2,3; Wang, Hui1,2,3; Gui, Yawei1,2,3; Dang, Ruochen1,2,3; Hu, Bingliang1,3; Wang, Quan1,3 |
出版日期 | 2023 |
会议日期 | 2022-12-18 |
会议地点 | Virtual, Online, China |
关键词 | pupil detection Infrared and visible light images Unet attention mechanism shape-prior loss data augmentation |
卷号 | 12558 |
DOI | 10.1117/12.2651952 |
英文摘要 | In recent years, pupil detection in human eye images or videos has played a key role in many fields. In the field of eye tracking, the position of the center of the pupil is a basic problem, and the error of pupil detection will be magnified in subsequent calculations, which will seriously affect the performance of eye tracking. In this paper, we propose to use the currently popular semantic segmentation network for pupil detection task. We first train the Unet architecture as a benchmark, then introduce two different attention modules into Unet, and compare with the benchmark network. The results show that our method has a higher detection rate within 1-15 pixel errors. We also added an ellipse fitting error term to the loss function of the network to further improve the network performance. The training of the model is done on the LPW dataset. Finally, we also investigate the effect of data augmentation on generalization performance, with the model trained on the LPW dataset and tested on the I-SOCIALDB dataset. Although data enhancement slightly reduces the detection rate of the model in the original data set, it can improve the generalization performance of the model. © COPYRIGHT SPIE. Downloading of the abstract is permitted for personal use only. |
产权排序 | 1 |
会议录 | AOPC 2022: Optical Spectroscopy and Imaging |
会议录出版者 | SPIE |
语种 | 英语 |
ISSN号 | 0277786X;1996756X |
ISBN号 | 9781510662308 |
源URL | [http://ir.opt.ac.cn/handle/181661/96364] |
专题 | 西安光学精密机械研究所_光学影像学习与分析中心 |
通讯作者 | Hu, Bingliang; Wang, Quan |
作者单位 | 1.Key Laboratory of Biomedical Spectroscopy of Xi an, Key Laboratory of Spectral Imaging Technology, Xi an Institute of Optics and Precision Mechanics (XIOPM), Chinese Academy of Sciences, Xi'an; 710119, China 2.University of Chinese Academy of Sciences, Beijing; 100049, China; 3.Key Laboratory of Spectral Imaging Technology, Xi an Institute of Optics and Precision Mechanics (XIOPM), Chinese Academy of Sciences, Xi'an; 710119, China; |
推荐引用方式 GB/T 7714 | Song, Wenhui,Wang, Hui,Gui, Yawei,et al. A pupil detection method based on Unet with attention module and shape-prior loss[C]. 见:. Virtual, Online, China. 2022-12-18. |
入库方式: OAI收割
来源:西安光学精密机械研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。