search

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

使用ONNXRuntime部署YOLOV7人脸+关键点检测，包含C++和Python两个版本的程序
50 stars 7 forks source link

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

Closed KunMengcode closed 1 year ago

KunMengcode commented 1 year ago

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

OrtSessionOptionsAppendExecutionProvider_CUDA(sessionOptions, 0);