Thank you for your sharing code.
I want to know the test time of a single picture on the CPU and GPU. So I used your test script run_basics.py and modified 2 parts
Only put a picture in the test folder (TestImages_AFLW2000)
Annotated lines 119 and 121 in api.py
https://github.com/YadiraF/PRNet/blob/master/api.py#L119.
Finally, the 0.23s on the CPU is similar to yours, but the 70ms on the GPU is very different from your result of 9.8ms.
Could you tell me what the reason may cause this?
Thank you for your sharing code. I want to know the test time of a single picture on the CPU and GPU. So I used your test script run_basics.py and modified 2 parts