• Laser Journal
  • Vol. 45, Issue 1, 99 (2024)
HU Qiaowei1, TAN Hong2, LIU Xinjuan1, HU Nan1, and FANG Erxi1,*
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less
    DOI: 10.14016/j.cnki.jgzz.2024.1.099 Cite this Article
    HU Qiaowei, TAN Hong, LIU Xinjuan, HU Nan, FANG Erxi. Design of lightweight pupil segmentation algorithm based onimproved Mobile-UNet[J]. Laser Journal, 2024, 45(1): 99 Copy Citation Text show less

    Abstract

    Eccentric photographic vision screening equipment is an important means of rapid detection of refractive state, and pupil image segmentation is an important part of its imaging algorithm. Aiming at the problems of limited computing resources and low precision of pupil segmentation in embedded devices, a lightweight pupil image segmentation algorithm based on improved Mobile-UNet was proposed. Based on U-Net improvement, the algorithm is preliminarily lightweight by using inverse residual linear bottleneck module. Group convolution is used to reduce parameters,channel mixing is used to open inter-group channels, and an adaptive parameter fusion parallel attention mechanism is introduced to improve segmentation performance. In addition, the optimization of the loss function enhances the attention to the boundary. The experimental results show that compared with MobilenetV2, the number of model parameters is reduced by 90%, the number of floating point operations is increased by 19%, but the segmentation performance is significantly improved. Compared with U-Net, the complexity of the model is greatly reduced and the segmentation performance is improved. Compared with other algorithms, this model has advantages in complexity and segmentation performance, and achieves lightweight and efficient segmentation.
    HU Qiaowei, TAN Hong, LIU Xinjuan, HU Nan, FANG Erxi. Design of lightweight pupil segmentation algorithm based onimproved Mobile-UNet[J]. Laser Journal, 2024, 45(1): 99
    Download Citation