hpc203/yolov7-detect-face-onnxrun-cpp-py

C++ uses GPU | 在C++中使用GPU

KunMengcode opened this issue · 0 comments

In YOLOV7_face Class Add the following code to the initialization location
在 YOLOV7_face 类中,初始化位置添加如下代码

OrtSessionOptionsAppendExecutionProvider_CUDA(sessionOptions, 0);