B站爆火的AI作畫,我給做成了離線版

其實這款軟件早就做好了,最近實在是沒什么素材寫文章了,就先把壓箱底的拿出來吧。
最近B站很火的一個項目就是AI作畫工具Disco Diffusion 5,這是一個基于AI各種模型于一體的項目,目前主要是文字描述生成圖片功能,做出來的畫非常有感覺。

這是個開源項目,所以我就打包成離線版了,只要顯卡好,本地就能跑。
01?軟件概覽
軟件名字叫做:AI作畫(Disco Diffusion 5)小工具離線版,基于GitHub上的開源項目Disco Diffusion。如下,軟件主界面,還是以操作簡單為主。

02?軟件體驗
軟件非常大,因為有很多離線模型,下載完成后,按照網(wǎng)盤里的word教程即可進行體驗。一、安裝
1、解壓pic_disco.zip,生成pic_disco目錄;
2、網(wǎng)盤里的models文件夾移動到pic_disco目錄中;?
3、vgg16-397923af.pth存到:C:\Users\Administrator\.cache\torch\hub\checkpoints 注:可能目錄前綴(有的是C:\Users\Administrator,有的是C:\Users\User)不一樣,但都是用戶目錄下,創(chuàng)建\.cache\torch\hub目錄即可;
4、打開軟件 進入軟件目錄pic_disco,雙擊打開DD5.exe程序即可,軟件界面如下所示:

二、作圖配置
主目錄pic_disco文件夾的ini.json就是作圖配置文件,可以打開方式選擇文本文檔打開。如下圖簡單說明幾個關(guān)鍵配置點:

1、輸出圖片目錄:pic_disco\images_out;
2、過程圖片目錄:pic_disco\progress.png,每幾個step(頻率可配置)更新一次圖片;
3、初始圖路徑:默認? "init_image": null 代表不用初始圖,想使用初始圖,將null改為圖片路徑,例如下面這樣(記得\用\\代替):
? "init_image": "F:\\PYQT5\\disco-diffusion-main\\test.png"。
三、顯卡要求
默認模型下,可能需要至少6GB顯存,以下為網(wǎng)友測試情況:
(1) RTX2060 6G獨顯,圖片尺寸256*512可行;
(2) RTX1070 8G獨顯,250steps耗時預(yù)估2小時,圖片尺寸1280*720;
(3) RTX2070S 8G獨顯,450steps耗時預(yù)估16分鐘,圖片尺寸960*448;
(4) RTX3090 24G獨顯,450steps耗時預(yù)估10分鐘,圖片尺寸1280*720。
四、常見錯誤
1、爆顯存下面這些都是圖片設(shè)置過大導(dǎo)致的爆顯存,或者6GB以下的顯卡:
(1) Unable to find a valid cuDNN algorithm to run convolution
(2) CUDA out of memory
2、黑圖
很多小伙伴生成的圖一直是黑的,這種貌似是10代及以下顯卡,可能顯卡比較舊,不支持
03??網(wǎng)友作品
如下看看網(wǎng)友用DD5生成的畫作吧,確實很驚艷:








04?小結(jié)
這款軟件確實很有意思,唯一的問題就是對顯卡要求高,最起碼得6G以上獨顯,想要高清720P,得12G獨顯左右,感興趣的可以體驗一下了,詳細的作圖參數(shù)教程多刷刷B站相關(guān)視頻即可。