一种基于多任务模型的停车位识别方法和装置与流程

技术特征：
1.一种基于多任务模型的停车位识别方法，其特征在于，所述方法包括：获取待识别停车位的停车场图像；利用目标神经网络模型对所述停车场图像中包含的各个对象进行目标检测，得到多个目标对象区域；利用所述目标神经网络模型对各个所述目标对象区域进行实例分割和特征提取，得到目标停车位入口信息；所述目标停车位为用于停车的空闲停车位；所述目标神经网络模型为多任务模型，采用基于分类和基于回归的多任务自适应损失函数训练得到；将所述目标停车位入口信息标注在所述停车场图像上，以指引车辆停车。2.根据权利要求1所述的方法，其特征在于，所述目标神经网络模型通过以下方法训练得到：将停车场样本图像数据集输入初始神经网络模型中，所述停车场样本图像中标注有真实对象区域和真实停车位入口信息；利用所述初始神经网络模型，对所述停车场样本图像中包括的各个对象进行目标检测，得到多个预测对象区域；利用所述初始神经网络模型，对各个所述预测对象区域进行全局特征提取和局部特征提取，得到预测停车位入口信息；确定所述预测对象区域和所述真实对象区域之间的差值，得到第一残差；确定所述预测停车位入口信息与所述真实停车位入口信息之间的差值，得到第二残差；利用预先确定的多任务自适应损失函数，确定所述第一残差和所述第二残差对应的目标残差；基于所述目标残差修改所述初始神经网络模型的参数，对所述初始神经网络模型继续进行训练，直到所述目标残差小于预设阈值，则训练结束，得到目标神经网络模型。3.根据权利要求2所述的方法，其特征在于，所述利用预先确定的多任务自适应损失函数，确定所述第一残差和所述第二残差对应的目标残差，包括：将所述第一残差和所述第二残差代入预先确定的多任务自适应损失函数中，所述多任务自适应损失函数为：其中，σ
i
为各个任务的测量噪声，l
i
(w)为各个任务对应的残差的二次方，为l
i
(w)的权重项，为正则项，所述权重项和所述正则项的大小成反比。4.根据权利要求2所述的方法，其特征在于，所述预测停车位入口信息包括预测入口角点方向和预测入口宽度，所述确定所述预测停车位入口信息与所述真实停车位入口信息之间的差值，包括：确定所述预测入口角点方向和所述真实停车位入口信息中的真实入口角点方向之间的方向差值；确定所述预测入口宽度和所述真实停车位入口信息中的真实入口宽度之间的宽度差
值；将所述方向差值和所述宽度差值加权融合，得到所述预测停车位入口信息与所述真实停车位入口信息之间的差值。5.根据权利要求4所述的方法，其特征在于，所述预测入口角点方向包括预测左角点方向和预测右角点方向，所述真实入口角点方向包括真实左角点方向和真实右角点方向，所述确定所述预测入口角点方向和所述真实停车位入口信息中的真实入口角点方向之间的方向差值，包括：确定所述预测左角点方向与所述真实左角点方向之间的左角点差值，并确定所述预测右角点方向与所述真实右角点方向之间的右角点差值；确定所述左角点方向的l1损失函数与所述左角点差值的乘积，并确定所述右角点方向的l1损失函数与所述右角点差值的乘积，分别得到第一乘积和第二乘积；确定所述第一乘积和所述第二乘积的和，得到初始方向差值；确定所述初始方向差值与预设约束项的和，得到所述预测入口角点方向和所述真实停车位入口信息中的真实入口角点方向之间的方向差值；其中，所述约束项用于约束所述预测左角点方向和所述预测右角点方向相近似。6.根据权利要求5所述的方法，其特征在于，所述预设约束项为横坐标差值的绝对值和纵坐标差值的绝对值的和，所述横坐标差值为所述预测左角点的横坐标与所述预测右角点的横坐标的差，所述纵坐标差值为所述预测左角点的纵坐标与所述预测右角点的纵坐标的差。7.一种基于多任务模型的停车位识别装置，其特征在于，所述装置包括：图像获取模块，用于获取待识别停车位的停车场图像；目标检测模块，用于利用目标神经网络模型对所述停车场图像中包含的各个对象进行目标检测，得到多个目标对象区域；实例分割模块，用于利用所述目标神经网络模型对各个所述目标对象区域进行实例分割和特征提取，得到目标停车位入口信息；所述目标停车位为用于停车的空闲停车位；所述目标神经网络模型为多任务模型，采用基于分类和基于回归的多任务自适应损失函数训练得到；标注模块，用于将所述目标停车位入口信息标注在所述停车场图像上，以指引车辆停车。8.根据权利要求7所述的装置，其特征在于，所述装置还包括模型训练模块，所述模型训练模块用于：将停车场样本图像数据集输入初始神经网络模型中，所述停车场样本图像中标注有真实对象区域和真实停车位入口信息；利用所述初始神经网络模型，对所述停车场样本图像中包括的各个对象进行目标检测，得到多个预测对象区域；利用所述初始神经网络模型，对各个所述预测对象区域进行全局特征提取和局部特征提取，得到预测停车位入口信息；确定所述预测对象区域和所述真实对象区域之间的差值，得到第一残差；确定所述预测停车位入口信息与所述真实停车位入口信息之间的差值，得到第二残
差；利用预先确定的多任务自适应损失函数，确定所述第一残差和所述第二残差对应的目标残差；基于所述目标残差修改所述初始神经网络模型的参数，对所述初始神经网络模型继续进行训练，直到所述目标残差小于预设阈值，则训练结束，得到目标神经网络模型。9.一种电子设备，其特征在于，所述电子设备包括处理器和存储器，所述存储器中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由所述处理器加载并执行以实现如权利要求1-6任一项所述的基于多任务模型的停车位识别方法。10.一种计算机可读存储介质，其特征在于，所述存储介质中存储有至少一条指令、至少一段程序、代码集或指令集，所述至少一条指令、所述至少一段程序、所述代码集或指令集由处理器加载并执行以实现如权利要求1-6任一项所述的基于多任务模型的停车位识别方法。

技术总结
本发明涉及了一种基于多任务模型的停车位识别方法和装置，该方法包括：获取待识别停车位的停车场图像；利用目标神经网络模型对所述停车场图像中包含的各个对象进行目标检测，得到多个目标对象区域；利用目标神经网络模型对各个所述目标对象区域进行实例分割和特征提取，得到目标停车位入口信息；所述目标神经网络模型为多任务模型，采用基于分类和基于回归的多任务自适应损失函数训练得到；将所述目标停车位入口信息标注在所述停车场图像上，以指引车辆停车。本方案将目标检测和特征提取两个任务集成在目标神经网络中，并采用多任务自适应损失函数可以有效的平衡任务间的梯度和权重，提高模型精度的同时不会增加模型算法的时间复杂度。时间复杂度。时间复杂度。


技术研发人员：马然
受保护的技术使用者：中汽创智科技有限公司
技术研发日：2022.08.02
技术公布日：2022/12/12