发明公开
- 专利标题: 一种基于强化学习的剪枝和量化的轻量化目标检测方法
-
申请号: CN202311623123.6申请日: 2023-11-30
-
公开(公告)号: CN117456335A公开(公告)日: 2024-01-26
- 发明人: 芮挺 , 王东 , 李宏伟 , 杨成松 , 殷勤 , 刘恂 , 唐建 , 史建军 , 郑南
- 申请人: 中国人民解放军陆军工程大学
- 申请人地址: 江苏省南京市秦淮区后标营路88号
- 专利权人: 中国人民解放军陆军工程大学
- 当前专利权人: 中国人民解放军陆军工程大学
- 当前专利权人地址: 江苏省南京市秦淮区后标营路88号
- 代理机构: 南京纵横知识产权代理有限公司
- 代理商 许婉静
- 主分类号: G06V10/82
- IPC分类号: G06V10/82 ; G06V10/25 ; G06N3/0464 ; G06N3/082
摘要:
本发明公开了一种基于强化学习的剪枝和量化的轻量化目标检测方法,包括如下步骤:获取待检测的图像数据,将待检测的图像数据输入至训练好的轻量化目标检测模型中,得到对应的目标检测结果;其中,轻量化目标检测模型的训练方法如下:获取原始模型;基于强化学习的深度确定性策略梯度的智能体对原始模型进行迭代更新的剪枝处理,得到剪枝模型;基于IAO量化法对剪枝模型的参数进行迭代更新,得到训练好的轻量化目标检测模型。降低了轻量化目标检测模型的计算量和参数量,提升了推理帧率,同时维持模型预测精度,便于模型部署到嵌入式设备或移动端设备上。