Commit 8ef38ce8f020b1defa21720fa7b69bfa31030978
1 parent
83d0a685
we only use BinaryProto and for some reason Caffe thinks we are using HD5F
Showing
1 changed file
with
1 additions
and
1 deletions
openbr/plugins/classification/caffe.cpp
| @@ -45,7 +45,7 @@ private: | @@ -45,7 +45,7 @@ private: | ||
| 45 | } | 45 | } |
| 46 | 46 | ||
| 47 | CaffeNet *net = new CaffeNet(model, caffe::TEST); | 47 | CaffeNet *net = new CaffeNet(model, caffe::TEST); |
| 48 | - net->CopyTrainedLayersFrom(weights.toStdString()); | 48 | + net->CopyTrainedLayersFromBinaryProto(weights.toStdString()); |
| 49 | FLAGS_minloglevel = google::ERROR; // Disable Caffe's verbose output after loading the first model | 49 | FLAGS_minloglevel = google::ERROR; // Disable Caffe's verbose output after loading the first model |
| 50 | return net; | 50 | return net; |
| 51 | } | 51 | } |