用Python脫掉圖片人物的褲子
作者|李博Garvin
轉載自CSDN部落格
效果演示
本文案例使用的是開源專案instagan,是一種比較新的gan模型建模原理,來自2019年ICLR的論文,下面看下效果對照:
(出於人道主義,會把人物的長褲脫掉然後換上短裙)
環境配置
首先玩這個模型需要以下前提條件:
有梯子
python3.6 版本
有GPU環境(因為原始碼是要求必須在GPU的Cuda環境下執行,如果沒有GPU的同學推薦用PAI裡面的DSWhttps://data.aliyun.com/product/learn ,比較便宜 )
執行以下流程:
(1)下載實驗程式碼:
https://github.com/sangwoomo/instagan
(2)下載pre-trained model,如果只是實驗就直接用訓練好的模型即可(如果只玩換褲子這個實驗,下載pants2skirt_mhp_instagan這個模型即可):
https://drive.google.com/drive/folders/1xb9rR21MhMVselc6HTmOr73WOkOviFmO
(3)安裝程式碼中的requirement.txt中的依賴包
最後把下載的model檔案放到程式碼檔案的根目錄下,目錄結構如下(藍色部分為模型資料夾,裡面是200_net_G_A.pth和200_net_G_B.pth):
完成以上步驟,整個環境就搭建好了。
使用模型做圖片轉換
注:這個專案的程式碼有很多hard code的邏輯,需要嚴格按照下面的做法執行才有可能跑通,包括所有檔案的命名。
(1)先設定需要轉換的圖片
在datasets目錄下新增一個test資料夾,構建如下的檔案格式
testA和testB存放需要轉換的原圖,類似於“效果演示中”穿著長褲的圖片,testA_seg和testB_seg需要存放mask圖片。mask檔案是轉換圖片中的褲子樣式,如下圖對應例項圖片左邊的女生褲子(如果想轉換其它部位,就不用我舉例子了吧):
mask圖片還需要與被轉換圖片命名一致,詳細規則參見原始碼中的datasets,若不一致會出現以下錯誤:https://github.com/sangwoomo/instagan/issues/5
(2)執行圖片轉換邏輯
在工程的根目錄下執行以下程式碼,使用工程下的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環境下1分鐘就可以執行完畢,程式碼執行完畢後在工程的results資料夾下就能看到轉換好的圖片了。
總結
整個專案的效果還是不錯的,隨便傳張圖片都能脫掉人物的褲子。不過可能程式碼還沒來得及優化,可能在實驗的過程中有很多坑,大家需要一定的debug能力才能把工程跑起來,小白不太建議使用哈。
(本文為AI科技大本營轉載文章,轉載請聯絡原作者)
精彩推薦推薦閱讀:
點選“閱讀原文”,檢視作者更多精彩博文。
相關文章
- 深度學習脫掉圖片人物的褲子【python教程】深度學習Python
- topthink人物資訊中的封面圖片上傳。。。。
- 開發管理之教授的褲子分析
- 推薦一個去除圖片人物背景的工具RemovebgREM
- 用python刪除重複的圖片Python
- 用Python實現圖片的清晰掃描Python
- lxyt-415x 最想刪掉的圖片
- 用python來給圖片加水印Python
- 用python生成驗證碼圖片Python
- 用一張圖片,黑客就能黑掉你的電腦黑客
- python生成圖片Python
- 用Python寫一個圖片標註工具Python
- 羅永浩“讓人尿褲子”的產品釋出!
- 智慧穿戴裝置:穿上就變蜘蛛俠的褲子
- PS製作抽象人物線條GIF動態圖片效果抽象
- java 電子印章 png 透明圖片Java
- 用python下載自己在聚合圖床上傳的圖片的一種方法Python圖床
- Python 潮流週刊#71:PyPI 應該擺脫掉它的贊助依賴(摘要)Python
- python批量ppt轉圖片,pdf轉圖片,word轉圖片指令碼Python指令碼
- 用SVG的圖片格式如何劃入更改圖片的顏色?SVG
- Python應用開發——爬取網頁圖片Python網頁
- 【python】圖片插入文字Python
- Python 下載圖片Python
- Python批次裁剪圖片Python
- 電子商務圖片放大jqzoomOOM
- 用Laravel 給孩子買紙尿褲Laravel
- python九宮格圖片的原理Python
- 褲子換裙子,就問你GAN的這波操作秀不秀
- 機器學習框架ML.NET學習筆記【7】人物圖片顏值判斷機器學習框架筆記
- 用於Github的圖片資源Github
- Python批量處理圖片Python
- python多工抓取圖片Python
- Python《回車桌面圖片》Python
- python批量抓取美女圖片Python
- 造輪子之圖片預覽元件(preview)元件View
- 造輪子之圖片輪播元件(swiper)元件
- SmartGrip男士內褲:懶癌男夢寐以求的內褲
- 手動脫掉Asprotect的殼,(給初學者的) (9千字)