您所在的位置: 成果库 一种基于对比学习及YOLO-v5的目标检测模型改进装置及方法

一种基于对比学习及YOLO-v5的目标检测模型改进装置及方法

发布时间: 2023-08-01

来源: 试点城市(园区)

基本信息

合作方式: 合作开发
成果类型: 发明专利
行业领域:
制造业,计算机、通信和其他电子设备制造业,计算机视觉技术领域,网络结构
成果介绍
本发明公开了一种基于对比学习及YOLO‑v5的目标检测模型改进装置及方法,涉及计算机视觉技术领域,其技术方案要点是:包括YOLO‑v5网络结构与对比学习网络结构;对比学习网络结构用于计算并输出正、负样本图片之间的对比损失;YOLO‑v5网络结构通过卷积运算提取图像特征,在输出端生成预测框并设置非极大值抑制运算,保留预测框进行目标损失函数计算;计算出来的目标损失与对比学习网络的对比损失进行加权求和计算最终损失值,将最终损失值反向传回给YOLO‑v5网络进行参数更新。网络训练中加入了对比学习的思想,因此使训练出来的模型对于识别目标与背景差异更加明显,有效地提高了目标检测模型的检测精度。
成果亮点
训练出来的模型对于识别目标与背景差异更加明显,有效地提高了目标检测模型的检测精度。
团队介绍
中国民用航空飞行学院,简称“中飞院”,创建于1956年,是中国民航局直属的部省共建本科高校。学院作为国家全过程培养民航高素质人才的主力军、主渠道、主阵地,经过近70年的建设发展,形成了理、工、文、管、法、艺、教育多学科协调发展,以培养民航飞行员、工程技术和民航运行管理人才为主的办学格局,其中全民航70%的飞行员、80%的机长、90%的功勋飞行员从这里启航,被誉为“中国民航飞行员的摇篮”,成为世界民航职业飞行员培养规模最大、能力最强、安全最好、质量过硬,享誉国内外民航的高等学府。
成果资料