C# Emgu CV - 如何在 dnn 中使用 cuda

Question

public static void DetectObjects(Bitmap bitmapImage)
    {
        
        Image<Bgr, byte> image;

        image = bitmapImage.ToImage<Bgr, byte>();

        Mat input = DnnInvoke.BlobFromImage(image, 1 / 255.0, swapRB: true);

        _netModel.SetInput(input);

        Mat output = _netModel.Forward();

    }

    private static Net _netModel = DnnInvoke.ReadNetFromDarknet("B:/yolov3.cfg.txt", "B:/yolov3.weights");

项目有 EmguCV 版本 4.4.0.4099 和同版本的 CUDA 模块。

因此，我需要该函数在 GPU 上执行计算。

有多少人没有搜索过这个，我没有找到合适的信息。

Answer 1

用起来好像就够了

_netModel.SetPreferableBackend(Emgu.CV.Dnn.Backend.Cuda);

Source