From 8ef38ce8f020b1defa21720fa7b69bfa31030978 Mon Sep 17 00:00:00 2001 From: Josh Klontz Date: Tue, 26 Sep 2017 12:35:35 -0600 Subject: [PATCH] we only use BinaryProto and for some reason Caffe thinks we are using HD5F --- openbr/plugins/classification/caffe.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openbr/plugins/classification/caffe.cpp b/openbr/plugins/classification/caffe.cpp index 2c5ec63..183b9a5 100644 --- a/openbr/plugins/classification/caffe.cpp +++ b/openbr/plugins/classification/caffe.cpp @@ -45,7 +45,7 @@ private: } CaffeNet *net = new CaffeNet(model, caffe::TEST); - net->CopyTrainedLayersFrom(weights.toStdString()); + net->CopyTrainedLayersFromBinaryProto(weights.toStdString()); FLAGS_minloglevel = google::ERROR; // Disable Caffe's verbose output after loading the first model return net; } -- libgit2 0.21.4