?

在觀看本文之前，請答應(yīng)我要善良。昨天預(yù)告了下，發(fā)現(xiàn)很多同學(xué)對這個模型都表示出興趣，甚至有好多同學(xué)后臺發(fā)來照片讓我?guī)退麄兠撗澴印Ｊ谌艘贼~不如授人以漁，請這些同學(xué)好自為之~

01效果演示

本文案例使用的是開源項目instagan，是一種比較新的gan模型建模原理，來自2019年ICLR的論文，

下面看下效果對照：

(出于人道主義，會把人物的長褲脫掉然后換上短裙)

02環(huán)境配置

首先玩這個模型需要兩個前提條件：

有梯子

python3.6版本

有GPU環(huán)境（因為源代碼是要求必須在GPU的Cuda環(huán)境下運行，如果沒有GPU的同學(xué)推薦用PAI里面的DSWhttps://data.aliyun.com/product/learn?，比較便宜 ）

（1）下載實驗代碼：https://github.com/sangwoomo/instagan

（2）下載pre-trained model，如果只是實驗就直接用訓(xùn)練好的模型即可：https://drive.google.com/drive/folders/1xb9rR21MhMVselc6HTmOr73WOkOviFmO

（如果只玩換褲子這個實驗，下載pants2skirt_mhp_instagan這個模型即可）

（3）安裝代碼中的requirement.txt中的依賴包

最后把下載的model文件放到代碼文件的根目錄下，目錄結(jié)構(gòu)如下（藍(lán)色部分為模型文件夾，里面是200_net_G_A.pth和200_net_G_B.pth）：

完成以上步驟，整個環(huán)境就搭建好了。

03使用模型做圖片轉(zhuǎn)換

注：這個項目的代碼有很多hard code的邏輯，需要嚴(yán)格按照下面的做法執(zhí)行才有可能跑通，包括所有文件的命名。

（1）先設(shè)置需要轉(zhuǎn)換的圖片

在datasets目錄下新增一個test文件夾，構(gòu)建如下的文件格式

testA和testB存放需要轉(zhuǎn)換的原圖，類似于“效果演示中”穿著長褲的圖片，testA_seg和testB_seg需要存放mask圖片。mask文件是轉(zhuǎn)換圖片中的褲子樣式，如下圖對應(yīng)實例圖片左邊的女生褲子（如果想轉(zhuǎn)換其它部位，就不用我舉例子了吧）：

mask圖片還需要與被轉(zhuǎn)換圖片命名一致，詳細(xì)規(guī)則參見源代碼中的datasets，如果不一致會出現(xiàn)以下錯誤：https://github.com/sangwoomo/instagan/issues/5

（2）執(zhí)行圖片轉(zhuǎn)換邏輯

在工程的根目錄下執(zhí)行以下代碼，使用工程下的test.py這個測試代碼：

?python test.py --dataroot ./datasets/test --model instagan --name pants2skirt_mhp_instagan --loadSizeH 240 --loadSizeW 160 --fineSizeH 240 --fineSizeW 160 --ins_per 2 --ins_max 20

如果沒有報錯的話在GPU環(huán)境下1分鐘就可以執(zhí)行完畢，代碼執(zhí)行完畢后在工程的results文件夾下就能看到轉(zhuǎn)換好的圖片了。

04總結(jié)

整個項目的效果還是不錯的，不過可能代碼還沒來得及優(yōu)化，可能在實驗的過程中有很多坑，大家需要一定的debug能力才能把工程跑起來，小白不太建議使用哈。

?

原文鏈接
本文為云棲社區(qū)原創(chuàng)內(nèi)容，未經(jīng)允許不得轉(zhuǎn)載。

總結(jié)

以上是生活随笔為你收集整理的深度学习为图片人物换装【python代码教程】的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。