【机器学习】手把手用AI算法实现冰雪奇缘中的“冰雪魔法”
大家好,相信一定有很多的小伙伴看過動(dòng)漫電影《冰雪奇緣》~
《冰雪奇緣》講述的是在一個(gè)四面環(huán)海、風(fēng)景如畫的阿倫黛爾王國(guó),公主艾莎天生具有制造冰雪的超能力,將任何的場(chǎng)景轉(zhuǎn)眼變成“冰天雪地”般的特效,十分的神奇。
感覺的每次到夏天的時(shí)候看《冰雪奇緣》,簡(jiǎn)直有是夏季敗火的首選電影~
感覺接下來我不能接著劇透下去了,有興趣的一定要去看一看,迪士尼的動(dòng)漫電影的效果真的不錯(cuò)~
劇中公主艾莎的冰雪魔法的特效~
說個(gè)題外話,不知道大家看電影的時(shí)候是不是也一樣的有代入感呢?記得自己小時(shí)候看《蜘蛛俠》的時(shí)候,每次看到蜘蛛俠的電影的時(shí)候(來張劇照~)
看著蜘蛛俠能自由自在的蕩秋千,還能隨時(shí)隨地的從手腕里面吐出來蛛絲,
甚至也想被“蜘蛛”咬上一口...(哈哈哈,開個(gè)玩笑,畢竟那只是科幻電影中的特效)。
但是今天不一樣,我們今天使用Python語言以及深度學(xué)習(xí)模型算法來實(shí)現(xiàn)《冰雪奇緣》中的公主艾莎的冰雪魔法~
我們先直接看“魔法效果”~? ? ?
?
實(shí)現(xiàn)原理
在上面的幾組對(duì)比圖片中,我選取了幾組比較有代表性的效果圖片~
左邊的圖片為夏季的風(fēng)景圖片,右邊是經(jīng)過“冰雪魔法”處理后的圖片
哈哈哈,不賣關(guān)子了,這個(gè)冰雪魔法的真面目,就是深度學(xué)習(xí)模型中的CycleGAN,
GAN深度學(xué)習(xí)網(wǎng)絡(luò)模型是深度學(xué)習(xí)模型下的生成對(duì)抗網(wǎng)絡(luò)的范疇,一般來說,我們使用CycleGAN網(wǎng)絡(luò)來進(jìn)行圖像的風(fēng)格遷移。
事實(shí)上我們?cè)趦煞輬D片中所進(jìn)行的操作是圖片景色的“冬夏風(fēng)格遷移”,具體的實(shí)現(xiàn)原理是將夏季的景色圖片,或者冬季的景色圖片,進(jìn)入GAN網(wǎng)絡(luò)進(jìn)行模型的訓(xùn)練,提取特征參數(shù),進(jìn)行預(yù)測(cè)圖片的矩陣像素值運(yùn)算。
簡(jiǎn)單來理解,生成對(duì)抗網(wǎng)絡(luò)的深度學(xué)習(xí)模型,其實(shí)是生成器與判別器的一場(chǎng)博弈,生成器生成的圖片放入判別器進(jìn)行判別,模型性能最優(yōu)的時(shí)候是生成器生成的假圖片,判別器無法正確識(shí)別時(shí)候,性能最好。
算法實(shí)現(xiàn)步驟
Git下載代碼倉(cāng)庫
使用Git拷貝下來整個(gè)CycleGAN的Python代碼工程文件,代碼工程為使用pytorch的深度學(xué)習(xí)框架進(jìn)行編寫,Git倉(cāng)庫的地址:
https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
如果不想自己搭建深度學(xué)習(xí)的運(yùn)行環(huán)境,可以使用Google colab中的notebook
每個(gè)賬號(hào)大約有12小時(shí)的連續(xù)免費(fèi)算力使用~
顯卡配置為Tesla K80?14G顯存
在notebook終端執(zhí)行以下指令:
下載預(yù)訓(xùn)練模型
由于GAN類型的神經(jīng)網(wǎng)絡(luò)模型的訓(xùn)練時(shí)間比較長(zhǎng)
因?yàn)樾枰M(jìn)行卷積運(yùn)算操作進(jìn)行圖片的標(biāo)簽判別同時(shí)要進(jìn)行反卷積的運(yùn)算進(jìn)行“假圖片”的生成,因此網(wǎng)絡(luò)結(jié)構(gòu)較深,運(yùn)算參數(shù)復(fù)雜,訓(xùn)練時(shí)間較長(zhǎng)。
notebook終端執(zhí)行腳本命令,我們所使用的是冬夏風(fēng)格遷移的運(yùn)算模型,因此選擇summer2winter_yosemite的預(yù)訓(xùn)練模型
下載測(cè)試圖片數(shù)據(jù)集
我們所下載的測(cè)試圖片數(shù)據(jù)集中包含四個(gè)文件夾,trainA,trainB,testA,testB。
我們只需要使用其中的一個(gè)測(cè)試文件夾即可,我們選擇的模型是夏季風(fēng)格轉(zhuǎn)換為冬季風(fēng)格。
因此使用夏季風(fēng)格的測(cè)試圖片testA的文件中的圖片進(jìn)行預(yù)測(cè)結(jié)果~
在notebook終端中執(zhí)行腳本命令,下載冬夏圖片數(shù)據(jù)集文件。
執(zhí)行模型前向推理
準(zhǔn)備的資源條件已經(jīng)完善,接下來進(jìn)行測(cè)試,在終端執(zhí)行模型測(cè)試文件腳本,notebook中執(zhí)行命令:
顯示執(zhí)行完畢,在輸出終端輸出預(yù)測(cè)圖片結(jié)果日志,查看result文件夾下,已經(jīng)有圖片的處理結(jié)果,實(shí)驗(yàn)完成~
? ? ? ? ? ?
結(jié)語
? ?
怎么樣,你學(xué)會(huì)了嗎?趕快動(dòng)手試一試吧~
自己親手實(shí)現(xiàn)公主艾莎的“冰雪魔法”~
怎么樣,是不是效果很酷炫呢,我是千與千尋,我們下期見~
············END············
往期精彩回顧適合初學(xué)者入門人工智能的路線及資料下載機(jī)器學(xué)習(xí)及深度學(xué)習(xí)筆記等資料打印機(jī)器學(xué)習(xí)在線手冊(cè)深度學(xué)習(xí)筆記專輯《統(tǒng)計(jì)學(xué)習(xí)方法》的代碼復(fù)現(xiàn)專輯 AI基礎(chǔ)下載機(jī)器學(xué)習(xí)的數(shù)學(xué)基礎(chǔ)專輯溫州大學(xué)《機(jī)器學(xué)習(xí)課程》視頻 本站qq群851320808,加入微信群請(qǐng)掃碼: 與50位技術(shù)專家面對(duì)面20年技術(shù)見證,附贈(zèng)技術(shù)全景圖總結(jié)
以上是生活随笔為你收集整理的【机器学习】手把手用AI算法实现冰雪奇缘中的“冰雪魔法”的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Win11正版和盗版有什么区别
- 下一篇: 【NLP】AAAI21最佳论文Runne