计算机工程与应用2024，Vol.60Issue(2)：337-343,7.DOI:10.3778/j.issn.1002-8331.2210-0032
改进的语义分割模型及其应用
Improved Semantic Segmentation Model and Its Application
王耀文 1程军圣 1杨宇1
作者信息
- 1. 湖南大学 机械与运载工程学院,长沙 410082
- 折叠
摘要
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 archi-tecture search method with lower hardware requirements is proposed,which based on the differentiable neural architec-ture 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分类
信息技术与安全科学引用本文复制引用
王耀文,程军圣,杨宇..改进的语义分割模型及其应用[J].计算机工程与应用,2024,60(2):337-343,7.
