摘要
本发明涉及模型剪枝算法技术领域,尤其涉及一种基于剪枝算法的无人机目标检测模型轻量化方法,包括初始训练模型;稀疏训练;模型剪枝;模型微调;生成轻量化模型,本发明针对STUAV‑YOLO模型进行参数和结构的剪枝优化,移除神经网络中冗余、不重要的连接与权重,以提升模型运行效率,使其更适合部署于资源受限的无人机设备上,实现实时、准确的高空目标检测任务。
技术关键词
轻量化方法
模型剪枝
YOLO模型
参数
无人机设备
剪枝方法
算法技术
可读存储介质
策略
因子
连续性
复杂度
网络架构
基准
结构单元
精度
批量
计算机