Fixed arch=compute_53,code=[sm_53,compute_53] for Jetson TX1

AlexeyAB

2018-06-30 1e9d1fcedf1a361bcdb384f15b5b14bdb526576d

 src/detector.c

@@ -61,10 +61,14 @@
    srand(time(0));
    network net = nets[0];

   if ((net.batch * net.subdivisions) == 1) {
   const int actual_batch_size = net.batch * net.subdivisions;
   if (actual_batch_size == 1) {
      printf("\n Error: You set incorrect value batch=1 for Training! You should set batch=64 subdivision=64 \n");
      getchar();
   }
   else if (actual_batch_size < 64) {
         printf("\n Warning: You set batch=%d lower than 64! It is recommended to set batch=64 subdivision=64 \n", actual_batch_size);
   }

    int imgs = net.batch * net.subdivisions * ngpus;
    printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
@@ -87,7 +91,8 @@
    load_args args = {0};
    args.w = net.w;
    args.h = net.h;
    args.paths = paths;
   args.c = net.c;
   args.paths = paths;
    args.n = imgs;
    args.m = plist->size;
    args.classes = classes;
@@ -105,7 +110,7 @@
    args.hue = net.hue;

#ifdef OPENCV
   args.threads = 3;
   args.threads = 3 * ngpus;
   IplImage* img = NULL;
   float max_img_loss = 5;
   int number_of_lines = 100;
@@ -388,6 +393,7 @@
   load_args args = { 0 };
   args.w = net.w;
   args.h = net.h;
   args.c = net.c;
   args.type = IMAGE_DATA;
   //args.type = LETTERBOX_DATA;

@@ -482,7 +488,7 @@

   for (i = 0; i < m; ++i) {
      char *path = paths[i];
      image orig = load_image_color(path, 0, 0);
      image orig = load_image(path, 0, 0, net.c);
      image sized = resize_image(orig, net.w, net.h);
      char *id = basecfg(path);
      network_predict(net, sized.data);
@@ -595,6 +601,7 @@
   load_args args = { 0 };
   args.w = net.w;
   args.h = net.h;
   args.c = net.c;
   args.type = IMAGE_DATA;
   //args.type = LETTERBOX_DATA;

@@ -913,8 +920,18 @@
      int num_labels = 0;
      box_label *truth = read_boxes(labelpath, &num_labels);
      //printf(" new path: %s \n", labelpath);
      char buff[1024];
      for (j = 0; j < num_labels; ++j)
      {
         if (truth[j].x > 1 || truth[j].x <= 0 || truth[j].y > 1 || truth[j].y <= 0 ||
            truth[j].w > 1 || truth[j].w <= 0 || truth[j].h > 1 || truth[j].h <= 0) 
         {				
            printf("\n\nWrong label: %s - j = %d, x = %f, y = %f, width = %f, height = %f \n",
               labelpath, j, truth[j].x, truth[j].y, truth[j].w, truth[j].h);
            sprintf(buff, "echo \"Wrong label: %s - j = %d, x = %f, y = %f, width = %f, height = %f\" >> bad_label.list", 
               labelpath, j, truth[j].x, truth[j].y, truth[j].w, truth[j].h);
            system(buff);				
         }
         number_of_boxes++;
         rel_width_height_array = realloc(rel_width_height_array, 2 * number_of_boxes * sizeof(float));
         rel_width_height_array[number_of_boxes * 2 - 2] = truth[j].w * width;
@@ -973,6 +990,7 @@
   //for (i = 0; i < number_of_boxes; ++i)
   // printf("%2.2f,%2.2f, ", points->data.fl[i * 2], points->data.fl[i * 2 + 1]);

   printf("\n");
   float avg_iou = 0;
   for (i = 0; i < number_of_boxes; ++i) {
      float box_w = points->data.fl[i * 2];
@@ -996,8 +1014,8 @@
      float box_intersect = min_w*min_h;
      float box_union = box_w*box_h + anchor_w*anchor_h - box_intersect;
      float iou = box_intersect / box_union;
      if (iou > 1 || iou < 0) {
         printf(" i = %d, box_w = %d, box_h = %d, anchor_w = %d, anchor_h = %d, iou = %f \n",
      if (iou > 1 || iou < 0) { // || box_w > width || box_h > height) {
         printf(" Wrong label: i = %d, box_w = %d, box_h = %d, anchor_w = %d, anchor_h = %d, iou = %f \n",
            i, box_w, box_h, anchor_w, anchor_h, iou);
      }
      else avg_iou += iou;
@@ -1066,7 +1084,8 @@
{
    list *options = read_data_cfg(datacfg);
    char *name_list = option_find_str(options, "names", "data/names.list");
    char **names = get_labels(name_list);
   int names_size = 0;
   char **names = get_labels_custom(name_list, &names_size); //get_labels(name_list);

    image **alphabet = load_alphabet();
    network net = parse_network_cfg_custom(cfgfile, 1); // set batch=1
@@ -1075,6 +1094,11 @@
    }
    //set_batch_network(&net, 1);
   fuse_conv_batchnorm(net);
   if (net.layers[net.n - 1].classes != names_size) {
      printf(" Error: in the file %s number of names %d that isn't equal to classes=%d in the file %s \n", 
         name_list, names_size, net.layers[net.n - 1].classes, cfgfile);
      if(net.layers[net.n - 1].classes > names_size) getchar();
   }
    srand(2222222);
    double time;
    char buff[256];
@@ -1093,10 +1117,10 @@
            if(!input) return;
            strtok(input, "\n");
        }
        image im = load_image_color(input,0,0);
        image im = load_image(input,0,0,net.c);
      int letterbox = 0;
        //image sized = resize_image(im, net.w, net.h);
      image sized = letterbox_image(im, net.w, net.h); letterbox = 1;
        image sized = resize_image(im, net.w, net.h);
      //image sized = letterbox_image(im, net.w, net.h); letterbox = 1;
        layer l = net.layers[net.n-1];

        //box *boxes = calloc(l.w*l.h*l.n, sizeof(box));