defmain():
tflib.init_tf()
withopen('networks/generator_star-stylegan2-config-f.pkl', "rb") asf:
generator_network, discriminator_network, Gs_network=pickle.load(f)
w_avg=Gs_network.get_var('dlatent_avg')
noise_vars= [varforname, varinGs_network.components.synthesis.vars.items() ifname.startswith('noise')]
Gs_syn_kwargs=dnnlib.EasyDict()
Gs_syn_kwargs.output_transform=dict(func=tflib.convert_images_to_uint8, nchw_to_nhwc=True)
Gs_syn_kwargs.randomize_noise=FalseGs_syn_kwargs.minibatch_size=1truncation_psi=0.5face_latent=read_feature('results/generate_codes/0008.txt')
z=np.stack(face_latentfor_inrange(1))
tflib.set_vars({var: np.random.randn(*var.shape.as_list()) forvarinnoise_vars}) w=Gs_network.components.mapping.run(z, None)
w=w_avg+ (w-w_avg) *truncation_psi''' age.npy - 调整年龄 angle_horizontal.npy - 在左右方向上调整人脸角度 angle_vertical.npy - 在上下方向上调整人脸角度 beauty.npy - 调整颜值 emotion_angry.npy - 调整此项可增添/减弱一些生气的情绪(调整步幅建议缩小) emotion_disgust.npy - 调整此项可增添/减弱一些厌恶的情绪(调整步幅建议缩小) emotion_easy.npy - 调整此项可增添/减弱一些平静的情绪(调整步幅建议缩小) emotion_fear.npy - 调整此项可增添/减弱一些害怕的情绪(调整步幅建议缩小) emotion_happy.npy - 调整此项可增添/减弱一些开心的情绪(调整步幅建议缩小) emotion_sad.npy - 调整此项可增添/减弱一些伤心的情绪(调整步幅建议缩小) emotion_surprise.npy - 调整此项可增添/减弱一些惊讶的情绪(调整步幅建议缩小) eyes_open.npy - 调整眼睛的闭合程度 face_shape.npy - 调整脸型 gender.npy - 调整性别 glasses.npy - 调整是否戴眼镜 height.npy - 调整脸的高度 race_black.npy - 调整此项可接近/远离向黑种人变化 race_white.npy - 调整此项可接近/远离向白种人变化 race_yellow.npy - 调整此项可接近/远离向黄种人变化 smile.npy - 调整笑容 width.npy - 调整脸的宽度 '''direction_file='age.npy'coeffs= [-15., -12., -9., -6., -3., 0., 3., 6., 9., 12.]
move_latent_and_save(w, direction_file, coeffs, Gs_network, Gs_syn_kwargs)