其他分享
首页 > 其他分享> > 虚拟人生成,Stylegan3 安装配置攻略

虚拟人生成,Stylegan3 安装配置攻略

作者:互联网

Stylegan是由英伟达(皮衣刀客老黄家的)实验室发布的虚拟人脸生成算法,可以无中生有,生成非常逼真的虚拟人,按现在流行的说法是数字人。AI通过学习大量的高清人脸图片之后,就能生成几十亿现实世界中不存在的人脸。

Stylegan1出现的时候已经让人惊掉下巴了,第三代自然是更加牛~,至于牛~在哪里?这个...只能意会不能言传。说人话就是升级了变强了,更加细腻逼真。具体改进可以看github主页,大概路径是stylegan->stylegan2->stylegan2-ada->stylegan3 。

算法交给大佬们,如何用交给我。

我之前非常详细的写过Stylgan的安装方法,目前也依旧可用。但是如果要在30系列跑起来就要费点周折了。今天就来写一下,在Rtx30上运行Stylgan3,生成一百张虚拟人脸。

基础准备

工欲善其事必先利其器,为了成功运行Styegan3,需要一些必备的软硬件基础设施。下载软件的时候注意软件的【版本号】,CUDA和cuDNN必须配套。

必备软件:

软件有点多,但是也不用太担心,基本都是下一步下一步安装。详细安装方法,可以参考我之前的文章《玩转StyleGan2:手把手教你安装并运行项目!》安装过程基本类似,注意区分版本即可。

必备硬件:

四肢健全的电脑一台

中高端N卡一张(我是3060 12G)

 

具体步骤

 

1. 获取源代码:

点击Windows开始图标,搜索mini,打开conda命令行工具。

输入命令:

E: cd dev 
git clone https://github.com/NVlabs/stylegan3.git

每输入一行命令,按一个回车,等待完成。我这里是把源代码放在了E盘下的DEV文件夹里,获取成功后会生成一个stylegan3的文件夹,里面的文件就是我们需要的东西。 这个步骤,一般来说很快,但是你极有可能被分在二班(谐音梗警告!)。

2. 设置环境变量

因为要用到VS,所以需要加一个环境变量。

环境变量路径:

C:\Program Files (x86)\Microsoft Visual Studio\<VERSION>\Community\VC\Auxiliary\Build\vcvars64.bat

这个环境变量加在Path下面,注意要替换成你本地环境的路径。如果你的VS是默认安装,那么主要是替换部分,根据我的版本来这里就是2019(尖括号不要留)。

3. 创建虚拟环境&安装依赖

官方已经准备好了一个environment.yml文件。里面有具体的依赖关系。直接运行相应命令就可以自动安装所有的依赖包

命令:

conda env create -f environment.yml 
conda activate stylegan3

第一行,创建环境,安装依赖

第二行,激活虚拟环境。

这一步完成,我以为万事俱备,就直接运行生成命了。没想到给了一个:

No CUDA runtime is found, using CUDA_HOME='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1'

问题是我已经安装过CUDA了,怎么说没有呢? 它是老大,只能按它说的。疯狂下载安装各种版本的CUDA,查各种牛头不对马嘴的解决方案。

最后...脑筋90度急转弯,我怀疑是Pytorch版本问题。

 

4. 安装最新版Pytorch

进入Pytorch官网

直接获取安装最新版的命令,怼上去!

pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

我的第一选择是使用conda命令安装,但是安装速度超级无敌慢,科学工具换了好几个都不行。最后用PIP命令以6M/s的速度安装成功。

上一步的问题真,的,解决,了。 针对这种牛头不对马嘴的错误提示,我无F可说,全凭悟性。目测是官方开发环境为linux,windows下加载包的时候没匹配到。

5. 运行命令生成图片

激动人心的时刻到了,开始创造虚拟人吧:

python gen_images.py --outdir=out --trunc=1 --seeds=2 --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl

没有想到的是....生成了....一条狗!一定是打开方式不对!

输入新的命令:

python gen_images.py --outdir=out --trunc=1 --seeds=2 --network=./models/stylegan3-r-ffhq-1024x1024.pkl

终于,生成了一个人,但是总感觉奇形怪状~效果好像还不如以前。

继续悟道,继续修改命令:

python gen_images.py --outdir=out --trunc=0.5 --seeds=1-100 --network=./models/stylegan3-r-ffhq-1024x1024.pkl

运行截图:

100张高清人脸,成功生成。图片会保存在stylegan3下面的out文件夹里面。

这里简单介绍一个下四个参数的含义。

--outdir 是图标输出路径

--trunc 这个取值范围0~1,值越变化越大。

--seed 控制图片生成的“种子” ,可以是单个数字,可以是一个区间。

--network 模型路径,可以是本地路径或者网络路径。

 

6. 再次运行

安装好之后,下次使用的步骤为:

1. 打开miniconda

2. 用CD命令进入指定目录

3. 用activate命令激活虚拟环境

3. 执行生成命令。

好了,搞定,收工,希望大家也能成功!

====

原文链接:Stylegan3 安装,配置,运行,生成! – 托尼是塔克

公众号:托尼不是塔克

标签:Stylegan3,--,虚拟人,生成,命令,CUDA,攻略,stylegan3,安装
来源: https://www.cnblogs.com/wangpg/p/15926556.html