-
公开(公告)号:CN118644672A
公开(公告)日:2024-09-13
申请号:CN202410768062.0
申请日:2024-06-14
Applicant: 杭州电子科技大学
IPC: G06V10/26 , G06V10/32 , G06V10/40 , G06V10/80 , G06V10/82 , G06V20/70 , G06V20/10 , G06V20/56 , G06N3/0464 , G06N3/048 , G06N3/045
Abstract: 本发明公开了一种基于轻量化注意力的城市街景实时实例分割方法,包括如下步骤:步骤1、读取城市街景数据集并进行预处理及数据增强;步骤2、经步骤1处理后的图像作为输入,通过骨干网络进行特征提取并输出特征图;步骤3、所述特征图输入至有聚焦函数的聚焦线性注意力模块进一步提取特征使其聚焦到目标实例上;步骤4、将所述聚焦注意力模块提取的特征图和骨干网络提取的不同尺寸的特征图进行融合;步骤5、基于解耦头得到检测分支以及掩码分支,将融合后的特征分别输入检测分支和掩码分支得到检测结果和分割掩码,最后将检测结果和分割掩码合成得到目标的实例分割结果。相对于传统的纯卷积神经网络和Transformer模型,它有效地平衡了性能和速度。