Invention Publication
- Patent Title: 一种基于YOLOv5与EWC的增量目标检测方法及系统
-
Application No.: CN202411853352.1Application Date: 2024-12-16
-
Publication No.: CN119762913APublication Date: 2025-04-04
- Inventor: 杨科 , 朱泳标 , 张广泽 , 李娜 , 唐兴荣 , 肖红玉 , 张营旭 , 王栋
- Applicant: 中铁二院工程集团有限责任公司 , 中国中铁股份有限公司
- Applicant Address: 四川省成都市金牛区通锦路三号;
- Assignee: 中铁二院工程集团有限责任公司,中国中铁股份有限公司
- Current Assignee: 中铁二院工程集团有限责任公司,中国中铁股份有限公司
- Current Assignee Address: 四川省成都市金牛区通锦路三号;
- Agency: 四川力久律师事务所
- Agent 胥莹
- Main IPC: G06V10/774
- IPC: G06V10/774 ; G06V10/25 ; G06V10/82 ; G06N3/0464 ; G06N3/0895

Abstract:
本发明公开了一种基于YOLOv5与EWC的增量目标检测方法及系统,检测方法包括以下步骤:S1、采用第一数据集对YOLOv5网络模型初始化后得到基模型;S2、利用基模型获取新数据集中的伪标签并更新新数据集,得到第二数据集;S3、采用正则化损失函数训练第二数据集并调整基模型的参数,得到第一模型;S4、更新第一模型的参数权重,记第一模型作为基模型,重复步骤S2~S3进行训练,直至参数调整后的第一模型满足训练结束条件,得到用于增量目标检测的网络模型。本发明使用的YOLOv5网络模型结合EWC算法,在保持高准确率的同时,检测和计算速度非常快在训练数据上表现良好,能很好地泛化到新的、未见过的新数据上,提高了目标检测的准确性和鲁棒性。
Information query