we only use BinaryProto and for some reason Caffe thinks we are using HD5F

Josh Klontz
1 parent 83d0a685
Showing 1 changed file with 1 additions and 1 deletions
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;
     }