Laser Journal, Volume. 45, Issue 1, 99(2024)

Design of lightweight pupil segmentation algorithm based onimproved Mobile-UNet

HU Qiaowei1... TAN Hong2, LIU Xinjuan1, HU Nan1 and FANG Erxi1,* |Show fewer author(s)
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Jul. 22, 2023

    Accepted: --

    Published Online: Aug. 6, 2024

    The Author Email: Erxi FANG (fangerxi@suda.edu.cn)

    DOI:10.14016/j.cnki.jgzz.2024.1.099

    Topics