AlexeyAB
2018-03-04 f39160f6e8f4465e9325e235e18f0d9413d1f672
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;
@@ -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];
            }