我正在尝试使用Caffe
导入我自己的预训练的Googlenet
OpenCV v.3.4.3
模型,我使用模型部署文件训练后运行Caffe
测试,一切正常。然而,当用图像blob喂食OpenCv
网(加载之后)后,我得到一个例外。
OpenCv代码:
Net net = Dnn.readNetFromCaffe("deploy.prototxt","bvlc_googlenet.caffemodel");
Mat image = Imgcodecs.imread(input.getAbsolutePath(), Imgcodecs.IMREAD_COLOR);
Mat blob = Dnn.blobFromImage(image);
System.out.println(image);
System.out.println(blob);
net.setInput(blob);
Mat result = net.forward().reshape(1);
输出错误:
Mat [ 24*15*CV_8UC3, isCont=true, isSubmat=false, nativeObj=0x1bcd0740, dataAddr=0x1a9d1880 ]
Mat [ -1*-1*CV_32FC1, isCont=true, isSubmat=false, nativeObj=0x1bcd0eb0, dataAddr=0x1a4e4340 ]
Exception in thread "main" CvException [org.opencv.core.CvException: cv::Exception: OpenCV(3.4.3) Z:\build tools\opencv-3.4.3\modules\dnn\src\layers\fully_connected_layer.cpp:73: error: (-215:Assertion failed) 1 <= blobs.size() && blobs.size() <= 2 in function 'cv::dnn::FullyConnectedLayerImpl::FullyConnectedLayerImpl'
]
at org.opencv.dnn.Net.forward_1(Native Method)
at org.opencv.dnn.Net.forward(Net.java:62)
at test.OpenCVTests.main(OpenCVTests.java:54)
Caffe-train-val-model.prototxt Caffe-deploy-model.prototxt
提前致谢!
这个问题在这里解决了:https://github.com/opencv/opencv/issues/12578#issuecomment-422304736
“在Caffe-train-val-model.prototxt中没有来自Caffe-deploy-model.prototxt的loss3 / classifier_retrain。如果你试图在Caffe中多次运行这个模型,你会得到相同输入的不同输出,因为Caffe填充随机错过了重量。“
版权所有(github.com/dkurt)