基于改進U-Net網(wǎng)絡(luò)的圖像混合畸變校正方法

摘要： 圖像幾何畸變校正是許多計算機視覺應(yīng)用的關(guān)鍵預(yù)處理步驟。當(dāng)前基于深度學(xué)習(xí)的幾何畸變校正方法主要解決圖像的單一畸變校正問題，為此，本文提出一種改進U-Net網(wǎng)絡(luò)的圖像混合畸變校正方法。首先，提出了一種構(gòu)建混合畸變的圖像數(shù)據(jù)集的方法，解決了訓(xùn)練數(shù)據(jù)集稀缺以及畸變類型單一的問題。其次，采用U-Net網(wǎng)絡(luò)結(jié)合空間注意力機制進行圖像特征提取及畸變坐標圖的重建，將圖像校正問題轉(zhuǎn)為畸變圖像的逐...