用AI給黑白老照片上色是如何實現的?
在改革開放40年之際,百度聯合新華社,發起了“給舊時光上色”活動,利用百度AI技術”煥彩”,還原老照片本來的色彩,把過去幾十年的歷史瞬間更鮮活地帶到每個人眼前。那麼將黑白照片上色的原理是什麼呢?百度方面進行了細緻的解答。
原來,在黑白相機時代,雖然彩色的景觀呈現在相片上是黑白的,但景觀顏色的不同,在黑白照片上呈現的灰度也是不同的。依據黑白相片灰度的不同,AI能大致判別物品的顏色。
但依據灰度判別顏色不一定十分準確,一些灰度上的細微差別,造成的結果差異可能失之毫釐,差之千里。比如藍色和綠色的灰度接近,但如果上色後的草成了藍色,牛仔褲成了綠色,那就大錯特錯了。
百度視覺技術部表示,“我們需要讓AI擁有常識,讓它可以判斷不同場景下的物品應該是什麼顏色。神經網絡就像一個簡化的大腦,你教給他東西,他就能記住東西,做出判斷。”
要做到這點,需要讓AI擁有圖形判別的能力。要讓AI能夠“認識”物品。百度“煥彩”項目的服務採用的是目前最為前沿的一種生成式對抗網絡技術。這項技術有兩個關鍵點:一是圖像生成器,二是對抗式學習方式。
所謂圖像生成器,就是當你給出一張黑白照片要生成彩色照片時,需要生成器博聞廣識,能準確的“回憶”起圖片裡面的事物原來是什麼顏色。
而對抗式學習方式則是難點中的難點。為了訓練生成器,百度又設計了新穎的判別器和損失函數來評價生成的彩色圖像和真實彩色圖像的接近程度。在學習過程中,生成器努力“記憶”事物的色彩分佈情況,使自己還原出來的彩色圖像越來越真實,讓判別器越來越難分辨。
在一個月的時間內,經過對萬餘張照片的學習,算法能力數次迭代,百度的AI技術終於能夠對黑白照片的圖像做出較為準確的判斷和上色。