search

zyainfal / One-Shot-Face-Swapping-on-Megapixels

One Shot Face Swapping on Megapixels.
Other
314 stars 40 forks source link

About the input of swapper? #19

Closed jo-dean closed 2 years ago

jo-dean commented 2 years ago

Hello,when i debug the inference.py, in the lineswapped_lats = self.swapper(idd_lats, att_lats) I found the dim of idd_lats and att_lats is [1,18,512]. but in the paper,_Lshigh and _Lthigh are the input of FTM, the dim is [1,14,512],did't include the _Ltlow

image Is there something different？ thank you!

zyainfal commented 2 years ago

Please check the forward function of swapper: https://github.com/zyainfal/One-Shot-Face-Swapping-on-Megapixels/blob/edfb9c85120fb9fd32386dca1465a70694b1207c/inference/megafs.py#L417-L435 where https://github.com/zyainfal/One-Shot-Face-Swapping-on-Megapixels/blob/edfb9c85120fb9fd32386dca1465a70694b1207c/inference/inference.py#L73 which means the lower 4 latents are not used by FTM but other 14 latents are used.

jo-dean commented 2 years ago

thank you! It's my fault.