摘 要:無人機進行目標檢測,檢測精度受圖像質量影響,圖像質量又受光照條件的影響。因此,設計了一種基于分類的輕量級無人機光適應算法———Adaptive Enhance and Exposure (AEE),利用輕量級網絡ShuffleNetv2 對航拍圖像按光照條件分類后,再利用改進的圖像處理模塊自適應光照轉換器(Illumination Adaptive Transformer,IAT) 進行圖像增強或者曝光校正;為了解決航拍低光圖像噪聲干擾問題,設計邊緣增強濾波模塊。實驗證明,輕量級網絡ShuffleNetv2 在自建航拍圖像分類數據集上分類精度達到99. 5% ;加入邊緣增強濾波模塊的圖像增強算法峰值信噪比(Peak Signal to Noise Ratio,PSNR)指標平均提升1. 26,結構相似性指數(Structural Similarity Index,SSIM) 指標提升0. 023;改進后的IAT 算法PSNR 指標提升0. 46,SSIM 指標提升0. 008;光適應算法對目標檢測平均準確率提升0. 4% ,在Jetson AGX Xavier 設備上處理速度達到57. 4 幀/ 秒,滿足實時目標檢測。
關鍵詞:分類;圖像增強;曝光校正;目標檢測;濾波
中圖分類號:TP389. 1 文獻標志碼:A
文章編號:1003-3106(2024)10-2425-09
0 引言
在無人機目標檢測和目標追蹤等任務中,圖像質量從根本上影響計算機視覺效果;同時,由于無人機資源受限,無法安裝使用性能特別優(yōu)良的攝像頭。因此,無人機條件下的計算機視覺任務中,往往會遇到諸多問題。一方面,在光照不足的情況下,圖像細節(jié)不足而且會受到相機內噪聲影響;另一方面,在某些場景中拍攝會遇到陽光或者強燈光直射導致的過/ 欠曝光問題。針對這些問題,現有的研究往往將其分為2 類任務:圖像增強和曝光校正。
圖像增強旨在微光條件下,增強圖像可視程度以及抑制噪聲的同時,恢復、保留圖像細節(jié)[1]。圖像增強的傳統(tǒng)算法有:基于直方圖均衡化的方法、基于Retinex 算法的方法等?;谏疃葘W習的算法有:結合Retinex 理論的RetinexNet 和KinD 兩種經典模型[2-3]、基于生成對抗網絡的ZeroDCE 網絡和Enlighten GAN 網絡等[4-5]。曝光校正旨在調整曝光不足和曝光過度的圖像,在短/ 長曝光條件下進行圖像重建[6-8]。曝光校正算法包括直方圖法[9]和深度學習法2 種,目前主流方法為基于計算機視覺的算法,例如Afifi 等[10]提出的基于深度神經網絡(DeepNeural Network,DNN)模型的曝光校正網絡以及在此基礎上,Nsampi 等[11]引入注意力機制設計的曝光校正網絡模型等。