How to use GPU?

[fjesikfjdskl]

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]

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

[fjesikfjdskl]

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

[610265158]

@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]

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