语义分割模型的部署方法、装置、存储介质及设备

本申请公开了一种语义分割模型的部署方法、装置、存储介质及设备，属于自动驾驶技术领域。创建适用于DLA的语义分割模型；获取训练集；对于每个训练样本，利用语义分割模型对样本图像进行处理，得到预测结果，利用细节损失函数、三个语义损失函数、边界损失函数和两个分割损失函数对预测结果、真实类别标签和真实边界标签进行损失计算，根据计算结果训练语义分割模型的模型参数；对训练好的语义分割模型进行INT8量化训练，以将语义分割模型由PyTorch格式转换为ONNX格式；将ONNX格式的语义分割模型部署到DLA上。本申请将语义分割模型从GPU迁移到DLA，在不显著牺牲精度的前提下，优化了语义分割模型的推理速度。