科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于网络剪枝和参数量化结合的遥感SAR目标检测方法,主要解决现有遥感SAR目标检测方法模型复杂度高和推理速度慢的缺陷。其实现方案为：从公开的遥感SAR目标检测数据集中获取划分好的训练集和测试集,对训练集进行数据扩充；对扩充后的训练集进行数据增强；对现有轻量化网络进行调整,构建基准模型；训练基准模型并计算其性能指标；评估调整后网络中每个特征提取模块的重要性,剪去不重要的模块,再进行滤波器剪枝；设计量化搜索空间,搜索剪枝后模型的混合精度量化方案,得到最终模型,使用该模型进行SAR目标检测。本发明在提高检测精度的同时节省了训练成本,可用于计算资源有限且实时性要求高的目标识别场景。
- 单位
