论文部分内容阅读
面向船舶自主航行对环境感知提出的需求,提升海上交通智能化水平,特别针对海面图像小尺度目标,如浮标、小渔船等通过海图、导航雷达难以探测的问题,提出一种基于改进YoloV5的海面小目标检测算法。首先,构建小目标数据集,针对小目标样本不均衡问题,对小目标样本进行数据增强;然后,对YoloV5网络进行改进,增加小目标检测层,获取更大的特征图;最后,完成改进YoloV5的网络训练,与YoloV5、YoloV4等方法进行对比实验,实验结果表明,提出方法能有效提高网络对小目标的检测能力,在受限水域场景下达到高召回率、高实时性的检测效果。