中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
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
DOI10.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收割

来源:西安光学精密机械研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。