a312863063 / generators-with-stylegan2 Goto Github PK
View Code? Open in Web Editor NEWHere is a series of face generators based on StyleGAN2
License: Other
Here is a series of face generators based on StyleGAN2
License: Other
您好,有幸看到您做的效果,实在太惊人,不知可否提供下您所写的训练代码呢?
本人邮箱:[email protected]
十分感谢该项目的分享,生成人脸质量很高。
我尝试下载了一些链接,发现 https://pan.baidu.com/s/1LabQMFLsKkYK3hLgRCQ-0A 链接挂了,请问还有备用链接吗?
期待您的回复!
想请教一下,人脸合成可以修改头发长短的属性吗,或者说怎么生成短发的图片
Hi @a312863063.
Your repos and the continuity in the work and datasets everything is awesome.
I have cloned your repo and taken your latent representations. I want to change pose positions of a human face horizontally, vertically (yaw, roll). I checked the results. The pose was changing. But also face structure is also changing. How can i get the face different poses (20 degrees rotation, 30 ...)without modifying face???
Thanks in Advance.
Regards,
SandhyaLaxmi
作者好,限于算力想下载现成模型,请问custom model在哪里呢
Hi there. Thanks for sharing this amazing project.
I'm curious about your process of making these specialized generators.
Did you use the original StyleGAN FFHQ network to train on a hand-picked data set ? Or did you train the network from scratch with the data images you collected from the Internet ? How did you gather those specialized data set ?
Also from what I saw from your research notes in your website, you talked about the combination of InfoGAN and StyleGAN. What's that about ?
Lastly, can you show me how you trained all those specialized latent directions ?
Many thanks.
作者你好,我们是浙江理工大学应用心理学系的本科生。我们正在做面孔的实验,可能需要使用到您提供的混血生成器,但是在你的仓库里面并没有找到这个生成器,不知道是否能提供一些混血的照片,用作于实验研究。
请问如果有多张gpu卡,怎么让它们一起跑呢
你好,请问有 https://models.com/ 网站上的数据集吗?
您好,我在允许main.py后出现了nvcc命令找不到的错误,请问是需要安装什么包吗 我是在ubuntu下面进行实验的
您好, 我在尝试还原您的整个神经网络用来训练还原图片的decoder。请问您在训练时使用的是最原本的nvlab的tf版本的stylegan2的网络嘛
比如男性、25岁、戴眼镜。。。
Numpy1.14.3 生成和编辑均不可使用
提示
(face) C:\Users\Administrator\AIFACE\generators-with-stylegan2-master>python main.py
ModuleNotFoundError: No module named 'numpy.core._multiarray_umath'
ImportError: numpy.core.multiarray failed to import
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "", line 968, in _find_and_load
SystemError: <class '_frozen_importlib._ModuleLockManager'> returned a result with an error set
ImportError: numpy.core._multiarray_umath failed to import
ImportError: numpy.core.umath failed to import
2020-09-04 16:46:22.413086: F tensorflow/python/lib/core/bfloat16.cc:675] Check failed: PyBfloat16_Type.tp_base != nullptr
Numpy1.19.1 可以生成,但编辑模式不行并有以下提示
sys:1: FutureWarning: arrays to stack must be passed as a "sequence" type such as list or tuple. Support for non-sequence iterables such as generators is deprecated as of NumPy 1.16 and will raise an error in the future.
(此版本一开始成功编辑过几次,突然跳出这提示,之后不管整个环境重装几次都是不行了,网上有人说是模型的问题,重新放模型也无法解决)
Numpy从1.14 ~1.19 都逐个安装了还是不行,1.6以下的提示都和1.14.3提示一样,超过1.6的提示都和1.19.1的提示一样
基于这个stylegan2,
linux环境有以下要求:
1.gcc版本不能高于9
2.gcc版本要至少符合cpp11特性
3.cuda只能是10.0
4.python37及以上应该是不支持的
分享我最后跑通的环境:
tf-gpu 1.14.0
gcc 7.3.1
python 3.6
cuda 10.0
cudnn 也要匹配 cuda
其实也可以考虑去dockerhub找找玩这个模型的人分享的容器镜像😀。
Mac 的A卡,搞不定cuda啊!
比如设为20,只有第一张是人脸,其它的全都是横状彩色波纹。其实第一张图片也有比较小的概率失败,不过从图案上看至少还能看到人脸的轮廓,其它的图就全是毫无意义的图案了。
加了点日志瞧了下,从运行时间上看,第一张图要15分钟(3070),剩下的图都是200毫秒内就结束了。
是否有其他人也遇到类似情况呢?
环境:3070, win10, anaconda python 3.7, vs2017. 脚本探测出来的GPU最高能力是sm_8x,nvcc不认识,手动降了档,试过75、72、61、62,都是一样的错误情况。
请问,你训练用了几块显卡,花了多少时间?
(stylegan2) hujinhong@xdcs:~/pc/generators-with-stylegan2-master$ python main.py Loading networks from "networks/generator_yellow-stylegan2-config-f.pkl"... Setting up TensorFlow plugin "fused_bias_act.cu": Preprocessing... Loading... Done. Setting up TensorFlow plugin "upfirdn_2d.cu": Preprocessing... Loading... Done. Generating image 0/20 ...
It's just stuck here. I can't make a picture
Generating image 0/20 ...
Traceback (most recent call last):
File "/opt/conda/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1356, in _do_call
return fn(*args)
File "/opt/conda/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1341, in _run_fn
options, feed_dict, fetch_list, target_list, run_metadata)
File "/opt/conda/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1429, in _call_tf_sessionrun
run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'FusedBiasAct' used by {{node Gs/_Run/Gs/G_mapping/Dense0/FusedBiasAct}}with these attrs: [gain=1.41421354, T=DT_FLOAT, axis=1, grad=0, alpha=0.2, act=3]
Registered devices: [CPU, XLA_CPU, XLA_GPU]
Registered kernels:
device='GPU'; T in [DT_HALF]
device='GPU'; T in [DT_FLOAT]
[[Gs/_Run/Gs/G_mapping/Dense0/FusedBiasAct]]
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "main.py", line 55, in
main()
File "main.py", line 52, in main
generate_images(network_pkl, generate_num)
File "main.py", line 39, in generate_images
images = Gs.run(z, None, **Gs_kwargs) # [minibatch, height, width, channel]
File "/home/aistudio/generators-with-stylegan2/dnnlib/tflib/network.py", line 442, in run
mb_out = tf.get_default_session().run(out_expr, dict(zip(in_expr, mb_in)))
File "/opt/conda/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 950, in run
run_metadata_ptr)
File "/opt/conda/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1173, in _run
feed_dict_tensor, options, run_metadata)
File "/opt/conda/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1350, in _do_run
run_metadata)
File "/opt/conda/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1370, in _do_call
raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'FusedBiasAct' used by node Gs/_Run/Gs/G_mapping/Dense0/FusedBiasAct (defined at :96) with these attrs: [gain=1.41421354, T=DT_FLOAT, axis=1, grad=0, alpha=0.2, act=3]
Registered devices: [CPU, XLA_CPU, XLA_GPU]
Registered kernels:
device='GPU'; T in [DT_HALF]
device='GPU'; T in [DT_FLOAT]
[[Gs/_Run/Gs/G_mapping/Dense0/FusedBiasAct]]
Errors may have originated from an input operation.
Input Source operations connected to node Gs/_Run/Gs/G_mapping/Dense0/FusedBiasAct:
Gs/_Run/Gs/G_mapping/Dense0/Const (defined at /home/aistudio/generators-with-stylegan2/dnnlib/tflib/ops/fused_bias_act.py:105)
Gs/_Run/Gs/G_mapping/Dense0/MatMul (defined at :46)
Gs/_Run/Gs/G_mapping/Dense0/mul_1 (defined at :67)
*research use only.
我安装了CUDA10.1、10.0及对应的CUDNN,首先切换到10.1的使用状态中,运行程序报类似GPU找不到的错误信息;
切换到10.0的使用状态中,又将numpy的版本从1.9降到和作者要求差不多的1.4.6后,以及VS2019后,GPU没问题了,运行报CUDA10.0支持VS最高到2017版本,我的是2019所以不支持;
更换到VS2017后,终于运行成功,出现了人脸合成图片。
感谢作者大大!
怎么让每次生成的图片不会覆盖掉前一次生成的图片
亲,您可以接商务合作吗?
用谷歌的colab构建了一下环境,自带tensorflow和gpu,可以在上面直接修改参数、查看效果
https://colab.research.google.com/drive/1iiTS5FyGZ0SbIg18nPzIWeeljPcciXkc?usp=sharing
@a312863063 希望您能优化下启动脚本,提供命令行传参功能
请问您的latent_directions是如何进行训练得到的?
可以公开一下那两个没有公开的的训练模型
1.黄金比例混血脸生成器(generator_mixed-blood-stylegan2-config-f.pkl)
2.亚洲美人脸生成器(generator_trendy-beauty-stylegan2-config-f.pkl)
现在安装的cuda是10.2的,安装tensorflow-gpu==1.14.0之后运行报错
ImportError: Could not find 'cudart64_100.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 10.0 from this URL: https://developer.nvidia.com/cuda-90-download-archive
更换1.15.0后报错
FileNotFoundError: [Errno 2] No such file or directory: 'networks/generator_yellow-stylegan2-config-f.pkl'
更换2.1.0后报错
ModuleNotFoundError: No module named 'tensorflow.contrib'
大佬能不能对cuda和tensorflow高版本做个适配,秋梨膏
作者的工作着实令人震撼,btw,能否请教以下如何编辑真实图片啊,我试了一些方案,迭代优化的w+时间太慢了要2分钟,并且控制表情太顿,笑得也不好看, 基于模型预测隐码的方式,重构图与原图不怎么像,怎么弄比较好,希望得到贵人的指导
How to train the network pkl file? 怎么自己训练模型呢? 训练的代码是在哪里?
Thanks very much!
你好,非常需要seeprettyface-face_editor里的camera_rotation.npy。谢谢
如题
请问有兴趣和计划升级到TF2吗?
anaconda环境
rtx3070
python 3.6.2
tensorflow 1.14
cudatoolkit 10.0.130
cudnn 7.6.5
查到30系需要cuda11以上,tensorflow也得一并升级
有没有30系显卡能跑这个项目的,麻烦交流一下
兄弟,你好:
不好意思在这开这个issue,看了你的博文:http://www.seeprettyface.com/research_notes_page3.html#intership
想试试训练OPN,无奈loss降不下去,能不能帮忙看看。。。
我用原始的10w+张网红数据集,基于style2官方源码训练框架,不管是在stylegan2-ffhq-config-f.pkl或者generator_wanghong-stylegan2-config-f.pkl模型上refine,很快清晰度就会下降,面部的立体感、发丝很明显不如generator_wanghong-stylegan2-config-f.pkl的效果,也调了下参数,感觉还是难以复现。能请教下其中的训练技巧吗
本人是心理学系的菜鸟,折腾了几天才跑通,分享一下自己的配置
WIN10 + RTX3060 + VS2019 + python3.7 + cuda11.2 + cudnn8.1.0 + tensorflow1.15
ANACONDA 虚拟环境下跑通。
坑1: 30系显卡 cuda版本必须高于11
坑2:11以上的cuda版本必须搭配2以上的tensorflow,但是这套代码不支持2以上版本tensorflow,因此这里用一个大神的魔改的tensorflow1.15可以跑通https://zhuanlan.zhihu.com/p/356526953
坑3: “无法打开包括文件: 'tensorflow/core/framework/op.h'”,参考这个链接可以解决https://blog.csdn.net/qq_39024280/article/details/111904417
坑4:annaconda虚拟环境下用conda命令安装cuda,无法使用nvcc,会报错,我不懂咋解决,就直接下载了11.2的安装包,全局安装。
Hi 想请问下为什么您生成的编辑方向,对不同的generator是同一套也能编辑呢?
我理解编辑方向对不同的生成器,应该是不同的会比较合理?不知道您git里的latent_directions是用哪个生成器做的,又为什么能用到其他生成器呢?
谢谢
How to generate latent directions on a finished network? Thanks a lot.
您好,非常感谢您的开源项目,从中学到了很多东西!
想请教在爬取得到图片后,需要使用 dlib 进行人脸的抠取,之后是否需要对人脸边框进行一定的扩大呢?根据您的经验,扩大比例为多少较为合适呢?
怎么更换成自己的背景图片呢?想要自己定制一套背景图片,要怎么配置呢?
请问你的网红脸生成器的训练集是通过FFHQ筛选得到的吗?有多少张?
Hi, thanks for the wonderful projects, I have a problem to download the models in the internet folder even using vpn. They always stop downloading after I downloaded about 200-300M. I was wondering if you could provide a baidupan links for those who are blocked by the great wall. Thanks a lot.
Xiaohong Zhao
A declarative, efficient, and flexible JavaScript library for building user interfaces.
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google ❤️ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.