一種基于知識蒸餾的圖像超分辨率增強方法

基本信息

申請?zhí)?/td> CN201810603516.3 申請日 -
公開（公告）號 CN108830813B 公開（公告）日 2021-11-09
申請公布號 CN108830813B 申請公布日 2021-11-09
分類號 G06T5/00（2006.01）I;G06T3/40（2006.01）I;G06T7/90（2017.01）I;G06N3/04（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 高欽泉;趙巖;童同 申請（專利權）人 福建帝視科技集團有限公司
代理機構 福州君誠知識產(chǎn)權代理有限公司 代理人 戴雨君
地址 350000福建省福州市鼓樓區(qū)西洪路528號云座2號樓5樓B區(qū)
法律狀態(tài) -

摘要

摘要 本發(fā)明公開一種基于知識蒸餾的圖像超分辨率增強方法，其包括以下步驟：1）訓練數(shù)據(jù)和測試數(shù)據(jù)的獲取；2）教師網(wǎng)絡的訓練；教師網(wǎng)絡擁有較深卷積層的神經(jīng)網(wǎng)絡模型，3）學生網(wǎng)絡的訓練；4）教師網(wǎng)絡對學生網(wǎng)絡的指導學習；通過三組指導實驗由學生網(wǎng)絡學習吸收教師網(wǎng)路的特征圖；5）測試評估圖像重建效果；6）根據(jù)輸出特征圖之間不同的矩陣關系對學生網(wǎng)絡進行進一步指導。本發(fā)明利用知識蒸餾的相關思想，將教師網(wǎng)絡的性能傳遞給學生網(wǎng)絡，學生網(wǎng)絡模型可以高效地運行在低功耗限制的移動設備和嵌入式設備上，學生網(wǎng)絡網(wǎng)絡結構不變的前提下，經(jīng)過教師網(wǎng)絡指導后的學生網(wǎng)絡的PSNR有顯著提升，得到更好的重建效果。