search

610265158 / Peppa_Pig_Face_Landmark

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

How to use GPU? #52

Open fjesikfjdskl opened 11 months ago

fjesikfjdskl commented 11 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 10 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 10 months ago

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

610265158 commented 10 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 10 months ago

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