DragGAN 已經(jīng)開源
地址:https://github.com/XingangPan/DragGAN
設(shè)備信息(筆記本):
系統(tǒng) ubuntu 18.4
顯卡 3080
步驟
1 clone 代碼到本地
git clone https://github.com/XingangPan/DragGAN
進(jìn)入到目錄里
cd DragGAN
2 下載預(yù)先訓(xùn)練的 StyleGAN2 權(quán)重
sh scripts/download_model.sh
3 使用 conda 創(chuàng)建虛擬python環(huán)境并啟用虛擬環(huán)境
創(chuàng)建環(huán)境
conda create -n DragGAN python=3.10
啟用環(huán)境
source activate DragGAN
4 安裝依賴
pip install -r requirements.txt
5 啟動(dòng)程序率翅,就可以愉快的玩耍了
sh scripts/gui.sh
界面還是比較簡(jiǎn)單搬素,操作也很容易上手
啟動(dòng)程序會(huì)提示依賴缺失問題万栅,我遇到的寫到下面,僅供參考
pip install imgui
pip install glfw
pip install pyOpenGL # 其實(shí)這個(gè)貌似不對(duì)
conda install -c anaconda pyopengl # 這個(gè)才對(duì)
pip install scipy