choyingw / SynergyNet

3DV 2021: Synergy between 3DMM and 3D Landmarks for Accurate 3D Facial Geometry
MIT License
383 stars 57 forks source link

Missing dependencies #10

Closed daviddavid closed 2 years ago

daviddavid commented 2 years ago

Additional python requirements that are needed, but not listed, are: cython and torchvision

Also, and more importantly, this project requires a cuda compatible GPU to run.

choyingw commented 2 years ago

Just added the mentioned libraries. You can comment out .cuda() and load the pretrained weights into CPU, if you don't have a GPU device.