def style_qinghua(pretrained=False):
model = _m(layer_count=9,
startf=16,
maxf=512,
truncation_psi=0.75,
truncation_cutoff=8,
mode='normal')
if pretrained:
checkpoint = 'https://github.com/GalAster/StyleGAN-Zoo/releases/download/v2.3.1/QingHua-1024x1024-649fe8.mat'
model.load_state_dict(_download(checkpoint, progress=True))
return model
def style_anime_face_d(pretrained=False):
model = _m(layer_count=8,
startf=32,
maxf=512,
truncation_psi=0.5,
truncation_cutoff=8,
mode='asuka')
if pretrained:
checkpoint = 'https://github.com/GalAster/StyleGAN-Zoo/releases/download/v2.0.0/AnimeFaceD-512x512-3e59ff.mat'
model.load_state_dict(_download(checkpoint, progress=True))
return model
def style_super_star(pretrained=False):
model = _m(layer_count=9,
startf=16,
maxf=512,
truncation_psi=0.75,
truncation_cutoff=8,
mode='normal')
if pretrained:
checkpoint = 'https://github.com/GalAster/StyleGAN-Zoo/releases/download/v1.7.0/SuperStar-1024x1024-4141b4.mat'
model.load_state_dict(_download(checkpoint, progress=True))
return model
def style_horo(pretrained=False):
model = _m(layer_count=8,
startf=32,
maxf=512,
truncation_psi=0.5,
truncation_cutoff=8,
mode='asuka')
if pretrained:
checkpoint = 'https://github.com/GalAster/StyleGAN-Zoo/releases/download/v1.1.0/Horo-512x512-822ee4.mat'
model.load_state_dict(_download(checkpoint, progress=True))
return model
def style_asian_people(pretrained=False):
model = _m(layer_count=9,
startf=16,
maxf=512,
truncation_psi=0.75,
truncation_cutoff=8,
mode='normal')
if pretrained:
checkpoint = 'https://github.com/GalAster/StyleGAN-Zoo/releases/download/v1.5.0/AsianPeople-1024x1024-82c02f' \
'.mat '
model.load_state_dict(_download(checkpoint, progress=True))
return model
def style_asuka(pretrained=False):
model = _m(layer_count=8,
startf=32,
maxf=512,
truncation_psi=0.5,
truncation_cutoff=8,
model='asuka')
if pretrained:
checkpoint = 'https://github.com/GalAster/StyleGAN-Zoo/releases/download/v1.0.0/Asuka-512x512.mat'
model.load_state_dict(
torch.hub.load_state_dict_from_url(checkpoint, progress=True))
return model
def style_asian_star(pretrained=False):
model = _m(layer_count=9,
startf=16,
maxf=512,
truncation_psi=0.75,
truncation_cutoff=8,
model='normal')
if pretrained:
checkpoint = 'https://github.com/GalAster/StyleGAN-Zoo/releases/download/v1.6.0/AsianStar-1024x1024.mat'
model.load_state_dict(
torch.hub.load_state_dict_from_url(checkpoint, progress=True))
return model
