模型训练、目标检测方法、装置、设备及存储介质

AITNT
正文
推荐专利
模型训练、目标检测方法、装置、设备及存储介质
申请号:CN202411661215
申请日期:2024-11-20
公开号:CN119559463A
公开日期:2025-03-04
类型:发明专利
摘要
本申请提出一种模型训练方法、目标检测方法、设备及存储介质,包括:将样本图像输入目标检测模型,输出包括目标对象的目标边框图;将目标边框图输入识别模型,获取目标边框图的预测分类结果以及预测掩膜;基于预测分类结果和预测掩膜,确定识别模型的损失函数值;基于损失函数值调整识别模型的模型参数,继续训练直至满足预设的训练完成条件,得到训练好的识别模型。本申请实施例通过基于预测分类结果和预测掩膜确定识别模型的损失函数值,以便在模型训练的过程中同时注意分类和掩膜提取的目标间辨别性强的轮廓特征,从而提高了识别模型的识别精准度。
技术关键词
图像 掩膜 模型训练方法 特征提取网络 样本 引入注意力机制 对象 标签 轮廓特征 处理器 可读存储介质 上采样 存储器 训练集 电子设备 参数 计算机 物体 程序
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号