why rotXYZ = self.rotXYZ.repeat(1, self.batch_size * 3, 1, 1) in function compute_rotation_matrix() when batch size > 1

LizhenWangT / FaceVerse

FaceVerse: a Fine-grained and Detail-controllable 3D Face Morphable Model from a Hybrid Dataset (CVPR2022)

BSD 2-Clause "Simplified" License

460 stars 57 forks source link

why rotXYZ = self.rotXYZ.repeat(1, self.batch_size * 3, 1, 1) in function compute_rotation_matrix() when batch size > 1 #5

Open leonardodora opened 2 years ago

leonardodora commented 2 years ago

in FaceVerseModel.py

there is error when batch_size > 1, how can I fix it？thanks！

LizhenWangT commented 2 years ago

This has been fixed in the newest version. Thank you for your message! But the codes in fit_images.py, tracking_online.py & tracking_offline.py are designed for batch=1. You still need to change the code depending on your purpose.