基于样本重构的船舶小目标检测算法研究
Research on tiny ship detection algorithm based on sample reconstruction作者机构:海南长光卫星信息技术有限公司海南海口570100 长光卫星技术股份有限公司吉林长春130102
出 版 物:《海洋测绘》 (Hydrographic Surveying and Charting)
年 卷 期:2024年第44卷第3期
页 面:78-82页
学科分类:083002[工学-环境工程] 0830[工学-环境科学与工程(可授工学、理学、农学学位)] 081802[工学-地球探测与信息技术] 08[工学] 0818[工学-地质资源与地质工程] 081602[工学-摄影测量与遥感] 0816[工学-测绘科学与技术]
基 金:国家重点研发计划(2021YFC3101802) 2022年海口市重点科技计划项目(2022-022)
摘 要:针对遥感影像船舶目标尺度小、场景分布不均匀、目标尺寸相对样本尺寸占比小以及深度学习模型对于小目标泛化性能不好的问题,提出一种样本重构的方法。首先对船舶目标按其最小外接矩形进行裁剪,然后采用多种方式合成标准尺寸样本,通过样本重构,可以提高样本中的目标占比,解决不同场景下目标分布不均匀的问题。实验发现,使用样本重构方法训练的模型,对于小目标的检测能力有所提升,结合在网络中添加小目标检测层,结果显示,模型在测试样本上的平均准确率(average precision, AP)从0.502提升到0.674,验证了该方法的有效性。