Xie Hanfei, Yuan Baoxi, Hu Chengyu, Gao Yujie, Wang Feng, Wang Yuqian, Wang Chunlan, Chu Peng
School of Electronic Information, Xijing University, Xi'an, China.
Xi'an Key Laboratory of High Precision Industrial Intelligent Vision Measurement Technology, Xijing University, Xi'an, China.
Front Neurosci. 2024 Sep 25;18:1471089. doi: 10.3389/fnins.2024.1471089. eCollection 2024.
Pathological myopia is a major cause of blindness among people under 50 years old and can result in severe vision loss in extreme cases. Currently, its detection primarily relies on manual methods, which are slow and heavily dependent on the expertise of physicians, making them impractical for large-scale screening. To tackle these challenges, we propose SMLS-YOLO, an instance segmentation method based on YOLOv8n-seg. Designed for efficiency in large-scale screenings, SMLS-YOLO employs an extremely lightweight model. First, StarNet is introduced as the backbone of SMLS-YOLO to extract image features. Subsequently, the StarBlock from StarNet is utilized to enhance the C2f, resulting in the creation of the C2f-Star feature extraction module. Furthermore, shared convolution and scale reduction strategies are employed to optimize the segmentation head for a more lightweight design. Lastly, the model incorporates the Multi-Head Self-Attention (MHSA) mechanism following the backbone to further refine the feature extraction process. Experimental results on the pathological myopia dataset demonstrate that SMLS-YOLO outperforms the baseline YOLOv8n-seg by reducing model parameters by 46.9%, increasing Box mAP@0.5 by 2.4%, and enhancing Mask mAP@0.5 by 4%. Furthermore, when compared to other advanced instance segmentation and semantic segmentation algorithms, SMLS-YOLO also maintains a leading position, suggesting that SMLS-YOLO has promising applications in the segmentation of pathological myopia images.
病理性近视是50岁以下人群失明的主要原因,在极端情况下可导致严重视力丧失。目前,其检测主要依靠人工方法,这些方法速度慢且严重依赖医生的专业知识,因此不适用于大规模筛查。为应对这些挑战,我们提出了SMLS-YOLO,一种基于YOLOv8n-seg的实例分割方法。为提高大规模筛查的效率而设计,SMLS-YOLO采用了极其轻量级的模型。首先,引入StarNet作为SMLS-YOLO的主干来提取图像特征。随后,利用StarNet中的StarBlock来增强C2f,从而创建C2f-Star特征提取模块。此外,采用共享卷积和降尺度策略来优化分割头,以实现更轻量级的设计。最后,该模型在主干之后引入多头自注意力(MHSA)机制,以进一步优化特征提取过程。在病理性近视数据集上的实验结果表明,SMLS-YOLO的模型参数减少了46.9%,Box mAP@0.5提高了2.4%,Mask mAP@0.5提高了4%,优于基线YOLOv8n-seg。此外,与其他先进的实例分割和语义分割算法相比,SMLS-YOLO也保持领先地位,这表明SMLS-YOLO在病理性近视图像分割中具有广阔的应用前景。