search

BadourAlBahar / pose-with-style

[SIGGRAPH Asia 2021] Pose with Style: Detail-Preserving Pose-Guided Image Synthesis with Conditional StyleGAN
https://pose-with-style.github.io/
MIT License
268 stars 41 forks source link

part flag doesn't work #11

Open Ibrahimatef opened 2 years ago

Ibrahimatef commented 2 years ago

Hi, Thank you for great work! I tried to specify the part to be transferred using --part with lower_body, or face, but it always transfers upper_body only as well as output region in red is upper body

fyviezhao commented 1 year ago

Same here. Moreover, can it transfer full_body outfits between persons, i.e., --part=full body?

fyviezhao commented 1 year ago

To enable the --part flag, modify this line:

g_ema = Generator(args.size, args.latent, args.n_mlp, channel_multiplier=args.channel_multiplier, garment_transfer=True).to(device)

to

g_ema = Generator(args.size, args.latent, args.n_mlp, channel_multiplier=args.channel_multiplier, garment_transfer=True, part=args.part).to(device)

I also create a pull request which supports full-body garment transfer.