如何在openCV中加载经过Caffe框架训练的神经网络?

时间:2018-10-22 14:39:12

标签: opencv import neural-network caffe mean

我想在OpenCV C ++应用程序中导入caffe神经网络模型(在Python中使用opencv也会遇到相同的问题)。神经网络是图像分类算法。

this link,我可以使用函数dnn::createCaffeImporter(modelTxt, modelBin);来做到这一点,其中modeltxt代表文件的路径,扩展名为*.caffemodel,其中包含使用caffe和{ {1}}是扩展名为modelBin的文件的路径,该文件描述了神经网络的体系结构。

但是,在输入要分类的图像之前先查看其他代码,它们会删除在训练和测试NN时从图像中删除的平均值。

显然,即使使用不同的相机并在不同的位置拍摄图像,使用神经网络时也必须删除相同的值。

在包含模型文件的文件夹中,存在另一个名为*.prototxt的文件。此文件包含要减去的方法

OpenCV是否提供一种直接加载此类文件的方法?奇怪的是,它提供了一种加载模型文件但不加载该文件的方法。否则,我应该安装Caffe框架只是为了加载三个值。

This是可从其中下载模型的页面。我正在查看的文件包含在子文件夹mean.binaryproto的归档文件CNRPark-Trained-Models.zip中。

0 个答案:

没有答案