add class_map

Hao0313 · Mar 2, 2017 · 41f15e0 · 41f15e0
1 parent 279841d
commit 41f15e0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/include/caffe/layers/region_loss_layer.hpp b/include/caffe/layers/region_loss_layer.hpp
@@ -8,6 +8,7 @@
 #include "caffe/proto/caffe.pb.h"
 #include <string>
 #include "caffe/layers/loss_layer.hpp"
+#include <map>
 
 namespace caffe {
 template <typename Dtype>
@@ -87,6 +88,9 @@ class RegionLossLayer : public LossLayer<Dtype> {
   Blob<Dtype> diff_;
   Blob<Dtype> real_diff_;
   tree t_;
+
+  string class_map_;
+  map<int, int> cls_map_;
 };
 
 }  // namespace caffe