HarDNet-Lite在嵌入式平台的优化方法
基本信息
申请号 | CN202011104925.2 | 申请日 | - |
公开(公告)号 | CN112232411A | 公开(公告)日 | 2021-01-15 |
申请公布号 | CN112232411A | 申请公布日 | 2021-01-15 |
分类号 | G06K9/62(2006.01)I;G06N3/04(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 黄文丽;杨省;高子昂;胡鹏;程金平;解伟荣 | 申请(专利权)人 | 苏州凌图科技有限公司 |
代理机构 | 西安通大专利代理有限责任公司 | 代理人 | 浙江凌图科技有限公司 |
地址 | 315099浙江省宁波市华楼巷15号 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种HarDNet‑Lite在嵌入式平台的优化方法,用以解决现有的目标检测网络过于复杂、运算量大、在嵌入式平台推理速度慢与定位精度低的问题。该方法包括以下步骤:1)搭建轻量级的HarDNet‑Lite特征提取网络;2)采用加权FPN结构融合不同尺度的特征图,使丰富的底层细节信息与高层的语义信息充分融合;3)生成YOLO检测头,在不同大小的特征图上面放置通过k‑means聚类生成的锚框,用以检测不同尺度大小的目标;4)使用分类与回归损失函数进行端到端的训练模型;5)将训练完成的模型部署于嵌入式平台,进行目标检测。本发明的有益效果是,提高了HarDNet‑Lite目标检测网络在嵌入式平台的推理速度与目标检测精度。 |
