search

610265158 / Peppa_Pig_Face_Landmark

A simple face detect and alignment method, which is easy and stable.
Apache License 2.0
525 stars 116 forks source link

How to use GPU? #52

Open fjesikfjdskl opened 9 months ago

fjesikfjdskl commented 9 months ago

Thank you very much for your open source, I want to ask how to use GPU to detect faces, looking forward to your reply!

610265158 commented 8 months ago

By default the onnx model use cpu to run. You can check onnx_model_base.py, to set work with gpu.

fjesikfjdskl commented 8 months ago

@610265158 Thank you very much for your recovery. After setting GPU, the detection speed becomes slow. Is this normal?

610265158 commented 8 months ago

@610265158 Thank you very much for your recovery. After setting GPU, the detection speed becomes slow. Is this normal?

I think there is some problem. It should not be slow. I didn't test the code with gpu.

fjesikfjdskl commented 8 months ago

Thank you very much, I will check the following details!