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

Question

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

Opened this issue 2 years ago · 1 comments

in FaceVerseModel.py

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

Answer 1 · 2022-05-17T14:12:36.000Z

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.