HarDNet-Lite在嵌入式平臺的優(yōu)化方法

基本信息

申請?zhí)?/td> CN202011104925.2 申請日 -
公開(公告)號 CN112232411A 公開(公告)日 2021-01-15
申請公布號 CN112232411A 申請公布日 2021-01-15
分類號 G06K9/62(2006.01)I;G06N3/04(2006.01)I 分類 計算;推算;計數;
發(fā)明人 黃文麗;楊省;高子昂;胡鵬;程金平;解偉榮 申請(專利權)人 蘇州凌圖科技有限公司
代理機構 西安通大專利代理有限責任公司 代理人 浙江凌圖科技有限公司
地址 315099浙江省寧波市華樓巷15號
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種HarDNet?Lite在嵌入式平臺的優(yōu)化方法,用以解決現有的目標檢測網絡過于復雜、運算量大、在嵌入式平臺推理速度慢與定位精度低的問題。該方法包括以下步驟:1)搭建輕量級的HarDNet?Lite特征提取網絡;2)采用加權FPN結構融合不同尺度的特征圖,使豐富的底層細節(jié)信息與高層的語義信息充分融合;3)生成YOLO檢測頭,在不同大小的特征圖上面放置通過k?means聚類生成的錨框,用以檢測不同尺度大小的目標;4)使用分類與回歸損失函數進行端到端的訓練模型;5)將訓練完成的模型部署于嵌入式平臺,進行目標檢測。本發(fā)明的有益效果是,提高了HarDNet?Lite目標檢測網絡在嵌入式平臺的推理速度與目標檢測精度。??