xiezhy6 / GP-VTON

Official Implementation for CVPR2023 paper "GP-VTON: Towards General Purpose Virtual Try-on via Collaborative Local-Flow Global-Parsing Learning"
475 stars 83 forks source link

Suspect an error in training code of train_warping.py #30

Open SHShim0513 opened 1 year ago

SHShim0513 commented 1 year ago

Dear authors,

Hello, First, thank you for sharing your great work!

While reading the implementation code, I think there is an error at line 118~123 of train_warping.py file.

I think that "part_mask" should be a concatenation of (data['person_clothes_left_mask'], data['person_clothes_middle_mask'], data['person_clothes_right_mask']). This is mainly because "part_mask" will be used as a mask at line 200, line 259, line 260, and line 261.

Can you check the code?

Thanks in advance.