added caffe-ssd demo

Author: jjykh

demo/caffe-ssd/demo.js

@@ -0,0 +1,42 @@
+"use strict";
+
+var caffe = require("../../bagua-caffe/build/Release/caffe.node");
+var SSD = require("./ssd.js");
+
+var LABELS = ["---", "aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow",
+    "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"];
+
+var image = caffe.io.loadImage("dog.jpg");
+
+caffe.io.namedWindow("Image");
+
+function show (blob) {
+  let detections = SSD.parse(blob, image);
+
+  detections.forEach(o => {
+    if (o.score > 0.3) {
+      console.log(LABELS[o.label], o.score.toFixed(3), "@", o.xmin, o.ymin, o.xmax, o.ymax);
+      image.rectangle(o.xmin, o.ymin, o.xmax, o.ymax);
+    }
+  });
+  caffe.io.show("Image", image);
+  caffe.io.waitKey();
+}
+
+
+var net = new caffe.Net('../../model-zoo/caffe/VOC0712/SSD_300x300/deploy.prototxt', '../../model-zoo/caffe/VOC0712/SSD_300x300/VGG_VOC0712_SSD_300x300_iter_120000.caffemodel');
+
+var transformer = new caffe.io.Transformer(net.blobs["data"].shape);
+transformer.setMeanValue([104, 117, 123]);
+
+net.blobs["data"].data = transformer.preprocess(image);
+
+// Sync call
+//show (net.forward()[0]);
+
+
+// Async call
+net.forward(output_blobs => {
+  show (output_blobs[0]);
+});
+

demo/caffe-ssd/dog.jpg

@@ -0,0 +1,23 @@
+exports.parse = function (blob, image) {
+  var w = image.cols, h = image.rows;
+  
+  var f = new Float32Array(blob.data.buffer);
+  var num_det = blob.shape[2];
+  var out = [];
+
+  for (var i = 0; i < num_det; i++) {
+    if (f[i * 7] >= 0) {
+      var o = {
+        label: f[i * 7 + 1] | 0,
+        score: f[i * 7 + 2],
+        xmin: (f[i * 7 + 3] * w) | 0,
+        ymin: (f[i * 7 + 4] * h) | 0,
+        xmax: (f[i * 7 + 5] * w) | 0,
+        ymax: (f[i * 7 + 6] * h) | 0
+      };
+      out.push(o);
+    }
+  }
+
+  return out;
+};