专利交易市场

本发明公开了一种基于YOLO V5算法的船舶多目标追踪方法，本发明首先对采集到的船舶图像数据进行筛选和标注并自组建数据集，将数据集划分为训练集、验证集、测试集；利用自组建数据集中的训练集和验证集对YOLO V5网络进行训练，获取基于YOLO V5网络的船舶检测模型以及权重文件；利用训练后的YOLO V5检测模型对测试集进行检测，输出检测结果，并对检测模型进行评价；基于训练后的YOLO V5检测模型，经DeepSort算法处理，生成追踪模型；对生成的DeepSort追踪模型进行实时性验证。本发明能够实现对海上船舶的检测及多目标跟踪，检测精度高、实时性好、速度快。