Yes.
you just need to change this line to your encoder to return W+ latent code of your face image, or directly load your W+ latent code to replace instyle:

DualStyleGAN/style_transfer.py

Lines 103 to 104 in 64285b1

    
           img_rec, instyle = encoder(F.adaptive_avg_pool2d(I, 256), randomize_noise=False, return_latents=True,  
        
                                   z_plus_latent=True, return_z_plus_latent=True, resize=False)

And then here set input_is_latent=True and z_plus_latent=False to indicate your instyle is W+ latent code rather than Z+ latent code.

DualStyleGAN/style_transfer.py

Lines 126 to 127 in 64285b1

    
           img_gen, _ = generator([instyle], exstyle, input_is_latent=False, z_plus_latent=True, 
        
                                 truncation=args.truncation, truncation_latent=0, use_res=True, interp_weights=args.weight)

	img_rec, instyle = encoder(F.adaptive_avg_pool2d(I, 256), randomize_noise=False, return_latents=True,
	z_plus_latent=True, return_z_plus_latent=True, resize=False)

	img_gen, _ = generator([instyle], exstyle, input_is_latent=False, z_plus_latent=True,
	truncation=args.truncation, truncation_latent=0, use_res=True, interp_weights=args.weight)

Changing the encoder #29

Description

Metadata

Metadata

Assignees

Labels

Projects

Milestone

Relationships

Development

Issue actions