ncnn使用vulkan加速 代码示例

ncnn是一个高效的神经网络前向计算框架，它支持使用Vulkan加速。下面是一个简单的代码示例：

#include <ncnn/gpu.h>
#include <ncnn/net.h>

int main()
{
    ncnn::create_gpu_instance();

    ncnn::Net net;
    net.load_param("squeezenet_v1.1.param");
    net.load_model("squeezenet_v1.1.bin");

    ncnn::Mat in = ncnn::imread("dog.jpg", ncnn::Mat::PIXEL_RGB);

    ncnn::Extractor ex = net.create_extractor();
    ex.set_vulkan_compute(true);
    ex.input("data", in);
    ncnn::Mat out;
    ex.extract("prob", out);

    ncnn::destroy_gpu_instance();
    return 0;
}

这段代码使用ncnn加载了一个预训练的squeezenet模型，并使用Vulkan加速对一张图片进行前向计算。需要注意的是，要使用Vulkan加速，需要在编译ncnn时启用Vulkan支持。