一种面向交通标志检测的小目标增强优化方法
摘要:
本发明公开了一种面向交通标志检测的小目标增强优化方法。本发明方法包含两部分的内容:采用基于优先度的小目标增强的策略进行数据增强。针对以往小目标增强无视各类型目标分布差异统一增强导致效果不佳的问题,采用基于优先度的小目标增强的策略进行数据增强,对小目标数据集进行针对性的增强。采用最优锚框宽高聚类优化训练数据。针对模型获取正样本时只关注比较大的样本、忽略小目标的问题,以及单纯以目标宽高进行聚类获取锚框初始值导致训练数据不合理的问题。采用最优锚框宽高聚类配合增强数据集优化训练数据。通过本发明提出的方法,最终可以提升小目标交通标志的检测精度以及整体检测精度。
0/0