基于双向层级自蒸馏的图像分割模型训练方法及相关装置
申请号:CN202511098187
申请日期:2025-08-06
公开号:CN120976232A
公开日期:2025-11-18
类型:发明专利
摘要
本发明公开了一种基于双向层级自蒸馏的图像分割模型训练方法及相关装置,包括:构建图像分割模型;使用Dice损失函数和KL散度双向蒸馏损失训练所述图像分割模型,得到训练后的图像分割模型,该方法及相关装置能够提高图像分割模型的精度。
技术关键词
图像分割模型
蒸馏
层级
神经网络架构
分支
中间层
训练系统
掩膜
可读存储介质
解码器
处理器
编码器
计算机设备
存储器
像素
模块
精度