你有過水下拍攝照片的經(jīng)歷嗎？你是否注意到水下拍攝的圖片總是有些模糊甚至失真？這是由于光的衰減和反向散射效應導致水下能見度較低。

為了解決這個問題，哈爾濱工程大學的科研人員設計了一種機器學習算法用于生成逼真的水下圖像，以及一種基于生成的圖像進行訓練從而實現(xiàn)自然色恢復和除霧的算法。團隊表示該算法在圖片處理質(zhì)量和數(shù)量上都與頂尖技術(shù)水平相當，該算法每秒在單個顯卡上處理的幀數(shù)可達125幀。

圖：圖像增強前后的水下目標探測結(jié)果。（A）真實的水下照片和（B）基于模型輸出的真實照片。紅色框內(nèi)為扇貝，藍色框內(nèi)為海參，綠色框內(nèi)為海膽。

團隊表示大多數(shù)水下圖像增強算法（例如調(diào)節(jié)白平衡算法）都不是基于物理成像模型構(gòu)建的，因此算法的泛化能力較差。而該技術(shù)采用的方法是首先利用生成對抗網(wǎng)絡（GAN）來生成一組特定調(diào)查地點的圖像，再將這組圖像輸入給第二個算法，U-Net。其中GAN是一種由生成器構(gòu)成的AI模型，用于使鑒別器在分類過程中將合成樣本歸類為真實樣本。

團隊運用3733個帶標簽的圖像以及對應的深度地圖訓練GAN，圖像主要包括扇貝、海參、海膽及其他水產(chǎn)養(yǎng)殖生物。他們還獲得了包括NY Depth在內(nèi)的公開數(shù)據(jù)集，該數(shù)據(jù)集總共包含數(shù)千張水下照片。

訓練完成后，研究人員將雙模型方法的結(jié)果與基線模型方法的結(jié)果進行對比。結(jié)果表明，該方法優(yōu)勢體現(xiàn)在兩方面，一方面是顏色還原均勻，另一方面是可以在不破壞原始輸入圖像的底層結(jié)構(gòu)的情況下很好地還原綠色色調(diào)的圖像。通常情況下，該方法還能夠在保持“適當?shù)摹绷炼群蛯Ρ榷鹊耐瑫r還原顏色，這是其他圖像增強方法所做不到的。

不過研究團隊基于受損圖片重建圖像的方法并非首創(chuàng)。劍橋咨詢公司的AI系統(tǒng)DeepRay用10萬個靜止圖片來訓練GAN網(wǎng)絡，用來消除由不透明玻璃板導致的圖片失真。開源的DeOldify計劃采用了包括GAN的一系列AI模型來對舊照片和膠卷進行著色和還原。

此外，微軟亞洲研究中心的科學家在去年九月詳細介紹了一個用于給視頻自動著色的端到端系統(tǒng)。去年英偉達的研究人員構(gòu)建了一個框架，該框架僅基于一個帶注釋的著色視頻幀就能推斷顏色。并且谷歌的AI團隊在去年六月推出了一種無需人工監(jiān)督即可對灰度視頻進行著色的算法。

責任編輯：gt