Update Readme.md

Alexey

2018-06-22 5de3a00496dd6e5c0e4c96694ae6b73f1373c6d0

 src/detection_layer.c

@@ -58,8 +58,8 @@
            int index = b*l.inputs;
            for (i = 0; i < locations; ++i) {
                int offset = i*l.classes;
                softmax_array(l.output + index + offset, l.classes, 1,
                        l.output + index + offset);
                softmax(l.output + index + offset, l.classes, 1,
                        l.output + index + offset, 1);
            }
        }
    }
@@ -285,3 +285,31 @@
}
#endif

void get_detection_detections(layer l, int w, int h, float thresh, detection *dets)
{
   int i, j, n;
   float *predictions = l.output;
   //int per_cell = 5*num+classes;
   for (i = 0; i < l.side*l.side; ++i) {
      int row = i / l.side;
      int col = i % l.side;
      for (n = 0; n < l.n; ++n) {
         int index = i*l.n + n;
         int p_index = l.side*l.side*l.classes + i*l.n + n;
         float scale = predictions[p_index];
         int box_index = l.side*l.side*(l.classes + l.n) + (i*l.n + n) * 4;
         box b;
         b.x = (predictions[box_index + 0] + col) / l.side * w;
         b.y = (predictions[box_index + 1] + row) / l.side * h;
         b.w = pow(predictions[box_index + 2], (l.sqrt ? 2 : 1)) * w;
         b.h = pow(predictions[box_index + 3], (l.sqrt ? 2 : 1)) * h;
         dets[index].bbox = b;
         dets[index].objectness = scale;
         for (j = 0; j < l.classes; ++j) {
            int class_index = i*l.classes;
            float prob = scale*predictions[class_index + j];
            dets[index].prob[j] = (prob > thresh) ? prob : 0;
         }
      }
   }
}