AaronJackson / vrn

:man: Code for "Large Pose 3D Face Reconstruction from a Single Image via Direct Volumetric CNN Regression"
http://aaronsplace.co.uk/papers/jackson2017recon/
MIT License
4.52k stars 742 forks source link

Runing code in CPU mode... #21

Closed msiraj83 closed 6 years ago

msiraj83 commented 6 years ago

I try to run the code.. For this purpose i prepare envrioment for GPU and run the code.. I got the out of memory error as shown in the screen shot.. out_of_memory_error Than i try to run the code in CPU mood by setting the gpunum=-1.. When i run the code than i got another error... as show below.... cpu_mode_error

Can anybody help me how to run this code in CPU mode ???

Thanks

AaronJackson commented 6 years ago

Out of interest, what GPU are you using and how much video RAM does it have?

Unfortunately the face-alignment code cannot be run on the CPU as it was built using nngraph. Although, since the face-alignment code uses much less memory, perhaps you can run that on the GPU and then the reconstruction code on the CPU?

Hope that helps. Thanks for the screenshots, very useful.

msiraj83 commented 6 years ago

@AaronJackson sorry for late reply... MY GPU is NVIDIA GEFORCE and 2GB... But i follow your instruction it work for now... is the texture code is not included???

Thanks

AaronJackson commented 6 years ago

There is an option in the run.m file which allows you to enable the vertex colouring