update

Author: hopef

README.md

@@ -94,4 +94,174 @@ void test(){
     //...
     //out就是结果
 }
+```
+
+SSD训练
+```
+#include <cc_utils.h>
+using namespace cc;
+
+class SSDMyDataLayer : public SSDDataLayer{
+public:
+    SETUP_LAYERFUNC(SSDMyDataLayer);
+
+    SSDMyDataLayer(){
+        this->datum_ = createAnnDatum();
+        this->label_map_ = loadLabelMap("labelmap_voc.prototxt");
+    }
+
+    virtual ~SSDMyDataLayer(){
+        releaseAnnDatum(this->datum_);
+    }
+
+    virtual int getBatchCacheSize(){
+        return 3;
+    }
+
+    virtual void* getAnnDatum(){
+        if (!loadAnnDatum("00001.jpg", "00001.xml", 0, 0, 0, 0, true, "jpg", "xml", this->label_map_, this->datum_)){
+            printf("无法加载.\n");
+            exit(0);
+        }
+        return this->datum_;
+    }
+
+    virtual void releaseAnnDatum(void* datum){
+    }
+
+private:
+    void* datum_;
+    void* label_map_;
+};
+
+void main(){
+    installRegister();
+    INSTALL_LAYER(SSDMyDataLayer);
+
+    WPtr<Solver> solver = loadSolverFromPrototxt("solver.prototxt");
+    solver->net()->copyTrainedParamFromFile("VGG_ILSVRC_16_layers_fc_reduced.caffemodel");
+    //solver->Restore("models/blstmctc_iter_12111.solverstate");
+    solver->Solve();
+}
+
+prototxt的data层：
+layer {
+  name: "data"
+  type: "CPP"
+  top: "data"
+  top: "label"
+  include {
+    phase: TRAIN
+  }
+  cpp_param{
+    type: "SSDMyDataLayer"
+  }
+  transform_param {
+    mirror: true
+    mean_value: 104
+    mean_value: 117
+    mean_value: 123
+    resize_param {
+      prob: 1
+      resize_mode: WARP
+      height: 300
+      width: 300
+      interp_mode: LINEAR
+      interp_mode: AREA
+      interp_mode: NEAREST
+      interp_mode: CUBIC
+      interp_mode: LANCZOS4
+    }
+    emit_constraint {
+      emit_type: CENTER
+    }
+  }
+  data_param {
+    batch_size: 8
+  }
+  annotated_data_param {
+    batch_sampler {
+      max_sample: 1
+      max_trials: 1
+    }
+    batch_sampler {
+      sampler {
+        min_scale: 0.3
+        max_scale: 1.0
+        min_aspect_ratio: 0.5
+        max_aspect_ratio: 2.0
+      }
+      sample_constraint {
+        min_jaccard_overlap: 0.1
+      }
+      max_sample: 1
+      max_trials: 50
+    }
+    batch_sampler {
+      sampler {
+        min_scale: 0.3
+        max_scale: 1.0
+        min_aspect_ratio: 0.5
+        max_aspect_ratio: 2.0
+      }
+      sample_constraint {
+        min_jaccard_overlap: 0.3
+      }
+      max_sample: 1
+      max_trials: 50
+    }
+    batch_sampler {
+      sampler {
+        min_scale: 0.3
+        max_scale: 1.0
+        min_aspect_ratio: 0.5
+        max_aspect_ratio: 2.0
+      }
+      sample_constraint {
+        min_jaccard_overlap: 0.5
+      }
+      max_sample: 1
+      max_trials: 50
+    }
+    batch_sampler {
+      sampler {
+        min_scale: 0.3
+        max_scale: 1.0
+        min_aspect_ratio: 0.5
+        max_aspect_ratio: 2.0
+      }
+      sample_constraint {
+        min_jaccard_overlap: 0.7
+      }
+      max_sample: 1
+      max_trials: 50
+    }
+    batch_sampler {
+      sampler {
+        min_scale: 0.3
+        max_scale: 1.0
+        min_aspect_ratio: 0.5
+        max_aspect_ratio: 2.0
+      }
+      sample_constraint {
+        min_jaccard_overlap: 0.9
+      }
+      max_sample: 1
+      max_trials: 50
+    }
+    batch_sampler {
+      sampler {
+        min_scale: 0.3
+        max_scale: 1.0
+        min_aspect_ratio: 0.5
+        max_aspect_ratio: 2.0
+      }
+      sample_constraint {
+        max_jaccard_overlap: 1.0
+      }
+      max_sample: 1
+      max_trials: 50
+    }
+  }
+}
 ```