7d8f3f66495979f1bbd27205d422d673991709f2,GANs/stylegan2/generate.py,,main,#,98

Before Change

        print(f"using style noise seed {args.seed_mix} for layers {args.mix_after}-{num_layers}.")
        rnd = np.random.RandomState(args.seed_mix)
        z2 = rnd.randn(batch_size, 512)
        style_noises = [nn.Variable((batch_size, 512)).apply(d=z)
                        for _ in range(args.mix_after)]
        style_noises += [nn.Variable((batch_size, 512)).apply(d=z2)
                         for _ in range(num_layers - args.mix_after)]
    else:
        // no style mixing (single noise / style is used)

After Change

        print(f"using style noise seed {args.seed_mix} for layers {args.mix_after}-{num_layers}.")
        rnd = np.random.RandomState(args.seed_mix)
        z2 = rnd.randn(batch_size, 512)
        style_noises = [nn.NdArray.from_numpy_array(z)]
        style_noises += [nn.NdArray.from_numpy_array(z2)]
    else:
        // no style mixing (single noise / style is used)
        print(f"using style noise seed {args.seed} for entire layers.")
        style_noises = [nn.NdArray.from_numpy_array(z) for _ in range(2)]

    nn.set_auto_forward(True)
    nn.load_parameters("styleGAN2_G_params.h5")

Instances