Add solver accuracy plotting (fix BVLC#32)

alejandro-isaza · Aug 18, 2015 · 105ddb0 · 105ddb0
1 parent e1ee5ff
commit 105ddb0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/include/caffe/solver.hpp b/include/caffe/solver.hpp
@@ -58,6 +58,7 @@ class Solver {
   int current_step_;
   shared_ptr<Net<Dtype> > net_;
   vector<shared_ptr<Net<Dtype> > > test_nets_;
+  vector<Dtype> accuracies;
 
   DISABLE_COPY_AND_ASSIGN(Solver);
 };

diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp
@@ -10,6 +10,7 @@
 #include "caffe/util/io.hpp"
 #include "caffe/util/math_functions.hpp"
 #include "caffe/util/upgrade_proto.hpp"
+#include "caffe/util/plot_data.h"
 
 namespace caffe {
 
@@ -344,6 +345,9 @@ void Solver<Dtype>::Test(const int test_net_id) {
       loss_msg_stream << " (* " << loss_weight
                       << " = " << loss_weight * mean_score << " loss)";
     }
+    if (output_name == "accuracy") {
+      accuracies.push_back(mean_score);
+    }
     LOG(INFO) << "    Test net output #" << i << ": " << output_name << " = "
         << mean_score << loss_msg_stream.str();
   }
@@ -372,6 +376,7 @@ void Solver<Dtype>::Snapshot() {
   snapshot_filename = filename + ".solverstate";
   LOG(INFO) << "Snapshotting solver state to " << snapshot_filename;
   WriteProtoToBinaryFile(state, snapshot_filename.c_str());
+  drawPNG(filename, accuracies);
 }
 
 template <typename Dtype>