针对嵌入式低算力平台的目标检测算法轻量化方法及系统

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
针对嵌入式低算力平台的目标检测算法轻量化方法及系统
申请号:CN202510053597
申请日期:2025-01-14
公开号:CN120147772A
公开日期:2025-06-13
类型:发明专利
摘要
本发明公开了针对嵌入式低算力平台的目标检测算法轻量化方法及系统,其中方法包括:对任务物体进行图像采集,得到样本多样化的数据集;选择Yolov8‑x模型作为教师模型,对Yolov8‑x模型的主干网络进行MIM自监督预训练,并对Yolov8‑x模型进行微调;选择MobileOne进行ConvModule模块替换作为学生模型,通过检测头交叉蒸馏法训练学生模型;使用结构化剪枝方法,对完成蒸馏学习训练的学生模型主干网络进行剪枝,并对完成剪枝的学生模型进行轻量化处理,经过轻量化处理的学生模型用于指定嵌入式平台部署。本发明解决了基于深度学习的目标检测算法计算资源有限、内存消耗过大以及高功耗的问题。
技术关键词
轻量化方法 学生 剪枝方法 网络 嵌入式平台 教师 蒸馏 算法 检测头 模型预训练 模型剪枝 重建误差 数据采集模块 优化器 卷积模块 监督学习方法 图像解码器