Commit 60803b165673814b2a6b4a7a15d84d9781292cb1
1 parent
5c4274be
turn off caffe verbose output
Showing
1 changed file
with
1 additions
and
1 deletions
openbr/plugins/classification/caffe.cpp
| @@ -44,9 +44,9 @@ private: | @@ -44,9 +44,9 @@ private: | ||
| 44 | Caffe::set_mode(Caffe::CPU); | 44 | Caffe::set_mode(Caffe::CPU); |
| 45 | } | 45 | } |
| 46 | 46 | ||
| 47 | + FLAGS_minloglevel = google::ERROR; // Disable Caffe's verbose output before loading any models | ||
| 47 | CaffeNet *net = new CaffeNet(model, caffe::TEST); | 48 | CaffeNet *net = new CaffeNet(model, caffe::TEST); |
| 48 | net->CopyTrainedLayersFromBinaryProto(weights.toStdString()); | 49 | net->CopyTrainedLayersFromBinaryProto(weights.toStdString()); |
| 49 | - FLAGS_minloglevel = google::ERROR; // Disable Caffe's verbose output after loading the first model | ||
| 50 | return net; | 50 | return net; |
| 51 | } | 51 | } |
| 52 | }; | 52 | }; |