如何在openCV中使用Caffe框架加载CNN Trained模型来测试(预测/分类)新图像?

时间:2016-04-29 04:00:21

标签: opencv machine-learning deep-learning caffe conv-neural-network

在使用10k图像在caffe框架中训练lenet模型之后,我得到了包含权重和baises的模型lenet_iter_4000.caffemodel。我在caffe中做了预测测试图像分类,现在我想通过加载这个caffemodel来测试图像在OpenCV中进行分类,任何人都可以帮助我如何将caffe和OpenCV结合起来预测新图像。

1 个答案:

答案 0 :(得分:2)

OpenCV contrib包含一个名为dnn的模块可以用于此,它可以加载Caffe和Torch模型,而here是GoogleNet的教程,您可以轻松地对其进行调整使用另一个网络,代码基本相同。

替代方案是Cafee的来源中的classification.cpp示例,该示例使用OpenCV读取图像并使用Caffe处理它。