首页 > 解决方案 > 如何使用 GPU 在 opencv 中使用 DNN 前向函数进行锐化

问题描述

我已经使用 OpenCVForUnity 和 opencvforsharp 来实现 Enet 神经网络进行分割。我想加速它的转发功能,我如何使用我的 gpu 来加速它。

我用过setPreferableTarget(Dnn.DNN_TARGET_OPENCL);但是太慢了

var enetmodel = "enet-model.net";

enet = Dnn.readNet(enetmodel);
var file = "example_01.png";
var org = Imgcodecs.imread(file);

var blob = Dnn.blobFromImage(org, 1.0 / 255, new Size(1024, 512));
enet.setPreferableBackend(3);
enet.setPreferableTarget(Dnn.DNN_TARGET_OPENCL);

enet.setInput(blob);
Mat score = enet.forward();

所以我想问一些其他的方法来使用 nvidia 来加速转发功能。感谢您的关注。

标签: c#gpu

解决方案


推荐阅读