基于深度学习的水面漂浮物目标检测评估

摘要

在这篇文章中,我们提出了一个关于水面漂浮物的小型数据集,并分析了几种目标检测模型在数据集上的表现,包括Faster R-CNN,R-FCN和SSD.我们的目的是探究目标检测模型在检测水面漂浮物特别是非物体类别时的特性,并找出权衡精确度和速度后最适合于引导水面清洁无人船的模型.为此,我们制作了一个小型的水面漂浮物数据集,数据集主要包括漂浮水草和漂浮落叶.之后我们通过将预训练模型在水面漂浮物数据集上进行迁移学习,实现了对于水面漂浮物区域的目标检测.我们对比并分析了这些模型的表现,SSD目标检测模型有着更高的精确度,Faster R-CNN模型则能给出更详细的预测,而同时拥有丰富结构特征和相当深度特征的模型对于困难目标有着更好的表现.