GithubHelp home page GithubHelp logo

mingtaoguo / draggan_pytorch Goto Github PK

View Code? Open in Web Editor NEW
93.0 4.0 11.0 157.33 MB

Unofficial implementation of DragGAN with StyleGAN2/3 pretrained models

License: MIT License

Dockerfile 0.15% Python 70.99% C++ 7.02% Cuda 21.84%
draggan gans stylegan2 stylegan3 image-editing

draggan_pytorch's People

Contributors

mingtaoguo avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar

draggan_pytorch's Issues

运行报错,求大神指导。代码有web交互的演示吗,我只找到了命令行运行

PS D:\DragGAN_pytorch-main> python draggan_stylegan2_human.py
Traceback (most recent call last):
File "D:\DragGAN_pytorch-main\draggan_stylegan2_human.py", line 165, in
draggan = DragGAN("stylegan2-human-1024x1024.pt", device="cuda")
File "D:\DragGAN_pytorch-main\draggan_stylegan2_human.py", line 101, in init
self.G.load_state_dict(torch.load(ckpt))
with _open_file_like(f, 'rb') as opened_file:
File "d:\anaconda3\lib\site-packages\torch\serialization.py", line 321, in _open_file_like
return _open_file(name_or_buffer, mode)
File "d:\anaconda3\lib\site-packages\torch\serialization.py", line 302, in init
super().init(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: 'stylegan2-human-1024x1024.pt'
PS D:\DragGAN_pytorch-main> python draggan_stylegan2_human.py
Setting up PyTorch plugin "bias_act_plugin"... Failed!
Traceback (most recent call last):
File "D:\DragGAN_pytorch-main\draggan_stylegan2_human.py", line 166, in
init_img = draggan.gen_image(seed=seed)
File "D:\DragGAN_pytorch-main\draggan_stylegan2_human.py", line 107, in gen_image
latent = self.G.mapping(z, label, truncation_psi=1.0, truncation_cutoff=None)
File "d:\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1502, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "d:\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1511, in _call_impl
return forward_call(*args, **kwargs)
File "D:\DragGAN_pytorch-main\training\networks_stylegan2_human.py", line 231, in forward
x = layer(x)
File "d:\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1502, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "d:\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1511, in _call_impl
return forward_call(*args, **kwargs)
File "D:\DragGAN_pytorch-main\training\networks_stylegan2_human.py", line 119, in forward
x = bias_act.bias_act(x, b, act=self.activation)
File "D:\DragGAN_pytorch-main\torch_utils\ops\bias_act.py", line 84, in bias_act
if impl == 'cuda' and x.device.type == 'cuda' and _init():
File "D:\DragGAN_pytorch-main\torch_utils\ops\bias_act.py", line 41, in _init
_plugin = custom_ops.get_plugin(
File "D:\DragGAN_pytorch-main\torch_utils\custom_ops.py", line 84, in get_plugin
raise RuntimeError(f'Could not find MSVC/GCC/CLANG installation on this computer. Check _find_compiler_bindir() in "{file}".')
RuntimeError: Could not find MSVC/GCC/CLANG installation on this computer. Check find_compiler_bindir() in "D:\DragGAN_pytorch-main\torch_utils\custom
ops.py".

机器学习小白一名,跑大神的包出了各种错,建了个学友们的交流学习互助群,答疑解惑交流提高,欢迎各位大神加入
微信图片_20230612092519

ERROR OCCURRED|UTC+8:00 2023/06/06 21:57

An error occurred when I tried this project for removing the watermark I added to the test image(a visible one, not the invisible one I referred to below). The snapshot of the Hugging Face website of this project is shown below.
网页捕获_6-6-2023_215816_huggingface co
(I covered the original image since there might be an invisible watermark containing my personal information on it.)
I'm not sure whether it is the fault of the project itself, the Internet or my PC. I just read the note on the website and decided to leave the issue here. I have only extremely poor knowledge of AI and my English is poor, too. If I had done something wrong and caused you trouble or wasted your time, I'm terribly sorry.
I sincerely hope your project will be better and better.

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.