计算机工程与应用 ›› 2024, Vol. 60 ›› Issue (2): 337-343.DOI: 10.3778/j.issn.1002-8331.2210-0032

• 工程与应用 • 上一篇    下一篇

改进的语义分割模型及其应用

王耀文,程军圣,杨宇   

  1. 湖南大学 机械与运载工程学院,长沙 410082
  • 出版日期:2024-01-15 发布日期:2024-01-15

Improved Semantic Segmentation Model and Its Application

WANG Yaowen, CHENG Junsheng, YANG Yu   

  1. College of Mechanical and Vehicle Engineering, Hunan University, Changsha 410082, China
  • Online:2024-01-15 Published:2024-01-15

摘要: 训练语义分割网络模型需要较为繁琐的人工标注作为训练标签,同时语义分割模型在构建和运行过程中也存在超参数较难确定以及模型过于庞大等问题。为解决这类问题,提出了一种基于标注框生成热点图的标签生成方法,简化了语义分割训练标签的人工标注过程。以及在可微分神经网络结构搜索方法的基础上提出了一种对硬件要求更低的神经网络结构搜索方法,并基于此种方法改进了特征金字塔结构,构建了一个改进的语义分割模型,并在安全帽与口罩检测数据集上进行了试验。与U-Net、FPN等模型比较,新的模型在参数量、计算速度以及精确度上都更有优势。

关键词: 语义分割模型, 神经网络结构搜索, 特征金字塔结构, 安全帽与口罩检测

Abstract: The training of semantic segmentation model requires complicated manual labeling, and there are also some problems in the construction and operation of semantic segmentation model, such as determining its hyperparameters and becoming bloated. To solve these problems, this paper proposes a label generation method based on heat map generated by ground truth box, which simplifies the manual labeling process of semantic segmentation training labels. A neural architecture search method with lower hardware requirements is proposed, which based on the differentiable neural architecture search method. By this method, the improved semantic segmentation model which contains a new feature pyramid is constructed. Tested on the helmet and mask detection datasets, compared with U-NET, FPN and other models, the new model takes the advantages in the number of parameters, calculation speed and accuracy.

Key words: semantic segmentation model, neural architecture search, feature pyramid structure, helmets and masks detection