《表3 不同anchor boxes下模型检测性能对比》
为了验证通过聚类取得训练集中船舶尺寸的先验信息是否能提升YOLOv2模型对船舶的检测性能,本实验对YOLOv2模型进行3组微调训练,分别采用YOLOv2的原始anchor boxes和聚类得到的两组anchor boxes,参数见表1。训练时batch size设置为16,为加速收敛选用冲量常数0.9,为防止过拟合选择权值衰减系数0.0005。训练30个迭代,前10个迭代的学习率为10-4,为促进模型优化后20个迭代的学习率降为10-5。在目标检测中精确率(precision)和召回率(recall)是一对互为矛盾的指标,因此本文选择平均精确率(average precision,AP)以及帧率(FPS)作为模型评价指标。采用不同anchor boxes的YOLOv2模型在测试集上的检测性能见表3。
图表编号 | XD00170296500 严禁用于非法目的 |
---|---|
绘制时间 | 2020.06.16 |
作者 | 段敬雅、李彬、董超、田联房 |
绘制单位 | 华南理工大学自动化科学与工程学院、华南理工大学自动化科学与工程学院、国家海洋局南海调查技术中心、华南理工大学自动化科学与工程学院 |
更多格式 | 高清、无水印(增值服务) |