首页 文章

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

提问于
浏览
2

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

1 回答

  • 2

    OpenCV contrib包含一个名为dnn的模块,可以用于此,它可以加载Caffe和Torch模型,而here是GoogleNet的教程,你可以轻松地将其改编为使用另一个网络,代码基本相同 .

    另一种选择是Cafee的源代码中的classification.cpp示例,它使用OpenCV读取图像并使用Caffe处理它 .

相关问题