search

cuiaiyu / dressing-in-order

(ICCV'21) Official code of "Dressing in Order: Recurrent Person Image Generation for Pose Transfer, Virtual Try-on and Outfit Editing" by Aiyu Cui, Daniel McKee and Svetlana Lazebnik
https://cuiaiyu.github.io/dressing-in-order
Other
513 stars 127 forks source link

Cannot reproduce results shown in the Colab demo notebook #30

Closed jackylu0124 closed 2 years ago

jackylu0124 commented 2 years ago

Hi there,

I tried running the demo notebook on both Colab and on my local machine but could not reproduce the results shown in the Colab demo notebook, here are some of the results from my end along with the original ones shown in the Colab notebook for comparisons. The fit looks correct in general, but the head and the skin tone are distorted and could not achieve the same level of fidelity as the ones shown in the original notebook. Do you by chance know why this might be happending and what I should do in order to reproduce the results you had in the demo notebook? Thank you very much in advance!

PyTorch version on Google Colab: 1.10.0+cu111 PyTorch version on local machine: 1.10.1+cu113

Original output: image My output: image

Original output: image My output: image

Original output: image My output: image

Original output: image My output: image

cuiaiyu commented 2 years ago

It's weird. Which checkpoints are you using? Is the opt.netG aligned with the checkpoints?

jackylu0124 commented 2 years ago

I am using the checkpoints in DIORv1_64.zip, it turns out I set opt.netG = "dior" earlier and forgot to set it as opt.netG = "diorv1" like you said. I can reproduce the exact same results now. Sorry for the inconvenience, but thank you so much for your help and hints!!