v/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2025-05-19 08:40:34 -07:00
Code Issues Projects Releases Wiki Activity

code convention

Browse Source
This commit is contained in:
Xintao 2021-05-18 20:56:37 +08:00
parent d9ec1351ff
commit b985478038
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
archs/arcface_arch.py
View File

@ -174,7 +174,7 @@ class ResNetArcFace(nn.Module):
layers = [] layers = []
layers.append(block(self.inplanes, planes, stride, downsample, use_se=self.use_se)) layers.append(block(self.inplanes, planes, stride, downsample, use_se=self.use_se))
self.inplanes = planes self.inplanes = planes
for i in range(1, blocks): for _ in range(1, blocks):
layers.append(block(self.inplanes, planes, use_se=self.use_se)) layers.append(block(self.inplanes, planes, use_se=self.use_se))
return nn.Sequential(*layers) return nn.Sequential(*layers)

4
archs/gfpganv1_arch.py
View File

@ -30,7 +30,7 @@ class StyleGAN2GeneratorSFT(StyleGAN2Generator):
num_style_feat=512, num_style_feat=512,
num_mlp=8, num_mlp=8,
channel_multiplier=2, channel_multiplier=2,
resample_kernel=[1, 3, 3, 1], resample_kernel=(1, 3, 3, 1),
lr_mlp=0.01, lr_mlp=0.01,
narrow=1, narrow=1,
sft_half=False): sft_half=False):
@ -231,7 +231,7 @@ class GFPGANv1(nn.Module):
out_size, out_size,
num_style_feat=512, num_style_feat=512,
channel_multiplier=1, channel_multiplier=1,
resample_kernel=[1, 3, 3, 1], resample_kernel=(1, 3, 3, 1),
decoder_load_path=None, decoder_load_path=None,
fix_decoder=True, fix_decoder=True,
# for stylegan decoder # for stylegan decoder