交通标志检测方法、装置、电子设备和存储介质

专利主权项内容

1.一种交通标志检测方法，其特征在于，所述方法包括：获取待测图像；基于第一网络提取所述待测图像的第一尺度图像特征；基于第二网络对所述待测图像进行上采样，得到所述待测图像的上采样结果，并基于所述待测图像的上采样结果提取所述待测图像的第二尺度图像特征；基于所述第二网络融合所述待测图像的第一尺度图像特征和所述待测图像的第二尺度图像特征，得到所述待测图像的多尺度图像特征，其中，所述待测图像的多尺度图像特征为所述待测图像的全局信息最优解；将所述多尺度图像特征作为交通标志检测头的输入，以基于所述交通标志检测头输出所述待测图像的特征图，其中，所述待测图像的特征图包括存在性特征、宽高特征、中心点偏移特征和交通标志类别特征，其中，所述宽高特征、中心点偏移特征基于中心点坐标特征确定，所述交通标志检测头预先基于训练集和测试集，以单通道学习方式学习提取所述中心点坐标特征，所述训练集包括样本和样本标注，所述样本标注包括坐标标注和存在性标注；基于所述存在性特征判断所述待测图像是否存在检测目标，如果存在所述检测目标，确定所述检测目标的位置索引和所述检测目标的坐标；基于所述交通标志类别特征确定所述检测目标的类别；基于所述检测目标的位置索引和所述中心点偏移特征确定所述检测目标的中心点偏移值；基于所述检测目标的位置索引和所述宽高特征确定所述检测目标的外接矩阵宽高。