avivga / overlord Goto Github PK

Thanks for your excellent work, have you done other attribute transfer? such as glasses without to with? thx

Did you have any artifacts during your training？As shown in the figure below, the highlight point or even colorful block on glasses, mouse, nose and ear. Any solutions?

Can you please provide the code for replicating the results of table 1 and table 2 in the paper?

Hi,

Thank you for the great work. I tried to do the inference on an image with resolution 1024*1024 using the following command
python main.py manipulate --model-name overlord-ffhq-x256-age --img face.png --output face_in_all_ages.png

but I get the following error:

Traceback (most recent call last): File "main.py", line 98, in <module> main() File "main.py", line 92, in main args.func(args) File "main.py", line 59, in manipulate manipulated_img = model.manipulate_by_labels(img) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context return func(*args, **kwargs) File "/home/userid2891/drive/projects/overlord/network/training.py", line 446, in manipulate_by_labels manipulated_img = self.amortized_model.generator(latent_code.unsqueeze(dim=0))[0] File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 167, in forward outputs = self.parallel_apply(replicas, inputs, kwargs) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 177, in parallel_apply return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)]) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 86, in parallel_apply output.reraise() File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/_utils.py", line 429, in reraise raise self.exc_type(msg) RuntimeError: Caught RuntimeError in replica 0 on device 0. Original Traceback (most recent call last): File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/parallel/parallel_apply.py", line 61, in _worker output = module(*input, **kwargs) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/userid2891/drive/projects/overlord/network/modules.py", line 76, in forward out = self.conv1(out, latent_code[:, 0]) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/userid2891/drive/projects/overlord/network/modules.py", line 139, in forward out = self.conv(input, style) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/userid2891/drive/projects/stylegan2-pytorch/model.py", line 258, in forward style = self.modulation(style).view(batch, 1, in_channel, 1, 1) File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "/home/userid2891/drive/projects/stylegan2-pytorch/model.py", line 158, in forward input, self.weight * self.scale, bias=self.bias * self.lr_mul File "/home/userid2891/anaconda3/envs/overlord/lib/python3.7/site-packages/torch/nn/functional.py", line 1753, in linear return torch._C._nn.linear(input, weight, bias) RuntimeError: mat1 dim 1 must match mat2 dim 0

Can you please guide me how can I solve this issue?
Also to download the CelebA dataset from google drive https://drive.google.com/open?id=0B7EVK8r0v71pWEZsZE9oNnFzTm8 it requires permission. Can you please make it public?
Thank you in advance.