Add solver accuracy plotting (fix BVLC#32)

include/caffe/solver.hpp

@@ -118,6 +118,7 @@ class Solver {
 
   // True iff a request to stop early was received.
   bool requested_early_exit_;
+  vector<Dtype> accuracies;
 
   DISABLE_COPY_AND_ASSIGN(Solver);
 };

src/caffe/solver.cpp

@@ -14,6 +14,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 {
 
@@ -411,6 +412,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();
   }
@@ -440,6 +444,19 @@ string Solver<Dtype>::SnapshotFilename(const string extension) {
   const int kBufferSize = 20;
   char iter_str_buffer[kBufferSize];
   snprintf(iter_str_buffer, kBufferSize, "_iter_%d", iter_);
+  filename += iter_str_buffer;
+  model_filename = filename + ".caffemodel";
+  LOG(INFO) << "Snapshotting to " << model_filename;
+  WriteProtoToBinaryFile(net_param, model_filename.c_str());
+  SolverState state;
+  SnapshotSolverState(&state);
+  state.set_iter(iter_);
+  state.set_learned_net(model_filename);
+  state.set_current_step(current_step_);
+  snapshot_filename = filename + ".solverstate";
+  LOG(INFO) << "Snapshotting solver state to " << snapshot_filename;
+  WriteProtoToBinaryFile(state, snapshot_filename.c_str());
+  drawPNG(filename, accuracies);
   return filename + iter_str_buffer + extension;
 }