-
公开(公告)号:CN118537534A
公开(公告)日:2024-08-23
申请号:CN202410600476.2
申请日:2024-05-15
Applicant: 哈尔滨工业大学(威海)
IPC: G06V10/25 , G06V10/774 , G06V10/82 , G06V20/05 , G06N3/0464 , G06N3/096 , G01S15/89 , G01V8/10 , G01S15/86
Abstract: 本申请属于水下灾害目标检测技术领域,具体为一种基于目标数据集与二次迁移学习的目标检测模型构建方法:创建声学‑光学水下灾害目标检测数据集;采用数据增强方法对声学‑光学水下灾害目标检测数据集进行扩展,同时,使用预训练模型权重在扩展过的声学‑光学水下灾害目标检测数据集随机提取部分形成的数据集上训练YOLOv5模型,将过程中最优权重作为第一阶段权重;采用第一阶段权重在扩展后的声学‑光学水下灾害目标检测数据集上对YOLOv5模型进行二阶段训练,获得水下灾害目标检测模型。本申请使用二次迁移学习和光学‑声学数据集结合构建的模型来检测水下灾害目标,提高了检测效率和准确性,降低现有模型检测水下灾害目标的难度。