基于自編碼器與YOLO算法的高清圖像小目標檢測方法

基本信息

申請?zhí)?/td> CN202010143805.7 申請日 -
公開（公告）號 CN111126359A 公開（公告）日 2020-05-08
申請公布號 CN111126359A 申請公布日 2020-05-08
分類號 G06K9/00;G06K9/62;G06N3/04;G06N3/08 分類 計算；推算；計數(shù);
發(fā)明人 吳憲云;孫力;李云松;王柯儼;劉凱;雷杰;郭杰;蘇麗雪;王康;司鵬輝 申請（專利權）人 南京億芯億意信息科技有限公司
代理機構 陜西電子工業(yè)專利中心 代理人 西安電子科技大學
地址 710071 陜西省西安市太白南路2號
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于自編碼器與YOLO算法的高清圖像小目標檢測方法，主要解決現(xiàn)有技術對高清圖像小目標檢測其準確率和速度不可兼顧的問題。實現(xiàn)步驟為：1)采集高清圖像并標注，獲得訓練集和測試集；2)對標注好的訓練集進行數(shù)據(jù)擴充；3)根據(jù)標注信息生成對應的Mask數(shù)據(jù)；4)搭建自編碼器模型；5)使用訓練集對其訓練；6)將訓練好的自編碼器的編碼網(wǎng)絡與YOLO?V3檢測網(wǎng)絡拼接，得到混合網(wǎng)絡并使用訓練集對其訓練；7)使用訓練好的混合網(wǎng)絡在測試集上進行目標檢測。本發(fā)明減少了目標檢測的計算量，提高了檢測速度，并在保證檢測速度的情況下提高了高清圖像中小目標的檢測精度，可用于無人機航拍圖像的目標識別。