标线识别方法和装置、标线识别模型训练方法、电子设备及程序产品

本申请公开一种标线识别方法，包括：将与标线相关的图像输入基于YOLO图像分割算法的标线识别模型进行标线识别处理，从而在图像中标识标线区域，具体包括：对输入图像进行特征提取，通过多个特征通道输出多个特征图；对多个特征通道添加部分注意力机制，从而对多个特征图的部分施加注意力处理，对多个特征图的其他部分不施加注意力处理；对经部分注意力处理的多个特征图进行采样处理，将经采样处理的多个特征图对应输出到多个分割头；利用多个分割头分别处理经采样处理的多个特征图，以分割提取多个特征图中的标线区域。本申请还涉及一种标线识别模型训练方法、标线识别模型、电子设备、程序产品。