多任务模型训练方法、图像处理方法、电子设备及存储介质
申请号:CN202510406186
申请日期:2025-04-02
公开号:CN120339691A
公开日期:2025-07-18
类型:发明专利
摘要
本发明的实施例提供了一种多任务模型训练方法、图像处理方法、电子设备及存储介质,涉及图像处理技术领域。该多任务模型训练方法包括:将裁切图像样本输入共用编码器,得到第一特征图,裁切图像样本包括标注分类标签、标注分割标签;将第一特征图输入分割网络,得到分割结果,并根据分割结果以及标注分割标签计算出分割损失,将第一特征图输入分类网络,得到分类结果,并根据分类结果、标注分类标签计算出分类损失,根据分类损失以及分割损失计算出多任务模型的综合损失,并利用综合损失更新多任务模型的参数。通过分割损失与分类损失的相互引导,从而提升模型的分类准确性。
技术关键词
多任务模型训练方法
分类网络
多尺度特征融合
注意力
图像处理方法
标签
嵌入特征
形态学特征
编码块
分类器
编码器
样本
融合特征
解码器
电子设备
图像处理技术
处理器