search

Gal4way / TPD

This is the official repository for the paper "Texture-Preserving Diffusion Models for High-Fidelity Virtual Try-On". CVPR 2024
116 stars 21 forks source link

Self-attention-based texture transfer #7

Closed cpis7 closed 5 months ago

cpis7 commented 6 months ago

Thanks for your great contribution! I found that you mentioned the self-attention-based texture transfer using concatenation of input on paper. But I'm struggling on finding the self-attention codes. Can you point out where did you concat the input and apply self-attention?

Gal4way commented 6 months ago

It's in the dataset.py. And we did not modify the code of self-attention, we just input a higher resolution image to the self-attention layer.