yoyo-nb / Thin-Plate-Spline-Motion-Model

[CVPR 2022] Thin-Plate Spline Motion Model for Image Animation.
MIT License
3.38k stars 552 forks source link

Equivariance loss #64

Open uestcMeng opened 1 year ago

uestcMeng commented 1 year ago

Thank you for sharing your work. I have some trouble in equivariance loss, which is Eq.10 in paper. In your code, it seems to trnsform the the key point of transformed frame, rather than transform the key point of driving frame. I want to know why it works or it is just a bug needed to fix?

https://github.com/yoyo-nb/Thin-Plate-Spline-Motion-Model/blob/9520e66fa5188f011b397f7de30fb5e79cdc47de/modules/model.py#L158