jswati31/cuda-ghr

Inference on one single image

Opened this issue · 2 comments

kevinkkk0321 commented

Hi @jswati31 , may I know how to run the pretrained model to do inference on one single image? thanks.

Thanks for your nice work!

jswati31 commented

Hello, thank you for your interest in our work. You can use these lines of code on your image, gaze, and head pose to run inference on one image:
https://github.com/jswati31/cuda-ghr/blob/master/train_cudaghr.py#L533-L558

kevinkkk0321 commented

Thanks for your reply! I'll test on the code.

May I ask if the model inference time can reach real-time?