Fixes for small objects. And fixes of densenet201_yolo.cfg, resnet50_yolo.c...

AlexeyAB

2018-03-08 a6c51e3b758aee7fd3a6f1d37daa8dcad4891e52

 src/data.c

@@ -304,9 +304,10 @@
    find_replace(labelpath, ".JPG", ".txt", labelpath);
    find_replace(labelpath, ".JPEG", ".txt", labelpath);
    int count = 0;
   int i;
    box_label *boxes = read_boxes(labelpath, &count);
   if (small_object == 1) {
      for (int i = 0; i < count; ++i) {
      for (i = 0; i < count; ++i) {
         if (boxes[i].w < 0.01) boxes[i].w = 0.01;
         if (boxes[i].h < 0.01) boxes[i].h = 0.01;
      }
@@ -316,7 +317,6 @@
    if(count > num_boxes) count = num_boxes;
    float x,y,w,h;
    int id;
    int i;

    for (i = 0; i < count; ++i) {
        x =  boxes[i].x;
@@ -326,7 +326,7 @@
        id = boxes[i].id;

      // not detect small objects
      if ((w < 0.01 || h < 0.01)) continue;
      if ((w < 0.001 || h < 0.001)) { printf("small w = %f, h = %f \n", w, h); continue; }

        truth[i*5+0] = x;
        truth[i*5+1] = y;
@@ -967,8 +967,8 @@
    for(i = 0; i < 10000; ++i){
        unsigned char bytes[3073];
        fread(bytes, 1, 3073, fp);
        int class = bytes[0];
        y.vals[i][class] = 1;
        int class_id = bytes[0];
        y.vals[i][class_id] = 1;
        for(j = 0; j < X.cols; ++j){
            X.vals[i][j] = (double)bytes[j+1];
        }
@@ -1031,8 +1031,8 @@
        for(i = 0; i < 10000; ++i){
            unsigned char bytes[3073];
            fread(bytes, 1, 3073, fp);
            int class = bytes[0];
            y.vals[i+b*10000][class] = 1;
            int class_id = bytes[0];
            y.vals[i+b*10000][class_id] = 1;
            for(j = 0; j < X.cols; ++j){
                X.vals[i+b*10000][j] = (double)bytes[j+1];
            }