search

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

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

Different Aspect Ratios #67

Open Trimad opened 1 year ago

Trimad commented 1 year ago

Is it possible to use this on images that are not perfectly square? I tried passing --img_shape 576,1024 but go this runtime error: RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 8 but got size 9 for tensor number 1 in the list.

goaldenboy commented 1 year ago

How can I match the frame size together for them to work