Opto-Electronic Engineering, Volume. 51, Issue 5, 240030(2024)

Real-time semantic segmentation algorithm based on BiLevelNet

Majing Wu1... Yong'ai Zhang1,2, Shanling Lin1,2, Zhixian Lin1,2, and Jianpu Lin12,* |Show fewer author(s)
Author Affiliations
  • 1School of Advanced Manufacturing, Fuzhou University, Quanzhou, Fujian 362200, China
  • 2Fujian Science & Technology Innovation Laboratory for Optoelectronic Information of China, Fuzhou, Fujian 350116, China
  • show less

    In response to the problem of the large parameter size of semantic segmentation networks, making it difficult to deploy on memory-constrained edge devices, a lightweight real-time semantic segmentation algorithm is proposed based on BiLevelNet. Firstly, dilated convolutions are employed to augment the receptive field, and feature reuse strategies are integrated to enhance the network's region awareness. Next, a two-stage PBRA (Partial Bi-Level Route Attention) mechanism is incorporated to establish dependencies between distant objects, thereby augmenting the network's global perception capability. Finally, the FADE operator is introduced to combine shallow features to improve the effectiveness of image upsampling. Experimental results show that, at an input image resolution of 512×1024, the proposed network achieves an average Intersection over Union (IoU) of 75.1% on the Cityscapes dataset at a speed of 121 frames per second, with a model size of only 0.7 M. Additionally, at an input image resolution of 360×480, the network achieves an average IoU of 68.2% on the CamVid dataset. Compared with other real-time semantic segmentation methods, this network achieves a balance between speed and accuracy, meeting the real-time requirements for applications like autonomous driving.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Majing Wu, Yong'ai Zhang, Shanling Lin, Zhixian Lin, Jianpu Lin. Real-time semantic segmentation algorithm based on BiLevelNet[J]. Opto-Electronic Engineering, 2024, 51(5): 240030

    Download Citation

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

    Category: Article

    Received: Jan. 30, 2024

    Accepted: Mar. 13, 2024

    Published Online: Jul. 31, 2024

    The Author Email: Lin Jianpu (林坚普)

    DOI:10.12086/oee.2024.240030

    Topics