Fixed replace_image_to_label(): replaces directories only for Pascal VOC an...

AlexeyAB

2018-05-22 a2666d5b3cfdcbab1e2eb66ca41f8b6835d4ffc8

 src/detector.c

@@ -129,9 +129,15 @@
         //int dim = (rand() % 12 + (init_w/32 - 5)) * 32;  // +-160
            //int dim = (rand() % 4 + 16) * 32;
         //if (get_current_batch(net)+100 > net.max_batches) dim = 544;
         int random_val = rand() % 12;
         int dim_w = (random_val + (init_w / 32 - 5)) * 32; // +-160
         int dim_h = (random_val + (init_h / 32 - 5)) * 32; // +-160
			
         //int random_val = rand() % 12;
         //int dim_w = (random_val + (init_w / 32 - 5)) * 32;  // +-160
         //int dim_h = (random_val + (init_h / 32 - 5)) * 32;  // +-160

         float random_val = rand_scale(1.4); // *x or /x
         int dim_w = roundl(random_val*init_w / 32) * 32;
         int dim_h = roundl(random_val*init_h / 32) * 32;

         if (dim_w < 32) dim_w = 32;
         if (dim_h < 32) dim_h = 32;

@@ -536,6 +542,7 @@
   char *mapf = option_find_str(options, "map", 0);
   int *map = 0;
   if (mapf) map = read_map(mapf);
   FILE* reinforcement_fd = NULL;

   network net = parse_network_cfg_custom(cfgfile, 1);   // set batch=1
   if (weightfile) {
@@ -714,9 +721,18 @@
               }
            }
         }
			
				
         unique_truth_count += num_labels;

         //static int previous_errors = 0;
         //int total_errors = fp_for_thresh + (unique_truth_count - tp_for_thresh);
         //int errors_in_this_image = total_errors - previous_errors;
         //previous_errors = total_errors;
         //if(reinforcement_fd == NULL) reinforcement_fd = fopen("reinforcement.txt", "wb");
         //char buff[1000];
         //sprintf(buff, "%s\n", path);
         //if(errors_in_this_image > 0) fwrite(buff, sizeof(char), strlen(buff), reinforcement_fd);

         free_detections(dets, nboxes);
         free(id);
         free_image(val[t]);
@@ -837,6 +853,7 @@
   free(truth_classes_count);

   fprintf(stderr, "Total Detection Time: %f Seconds\n", (double)(time(0) - start));
   if (reinforcement_fd != NULL) fclose(reinforcement_fd);
}

#ifdef OPENCV
@@ -1218,7 +1235,7 @@
         if(strlen(filename) > 0)
            if (filename[strlen(filename) - 1] == 0x0d) filename[strlen(filename) - 1] = 0;
        demo(cfg, weights, thresh, hier_thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename,
         http_stream_port, dont_show);
         http_stream_port, dont_show, ext_output);
    }
   else printf(" There isn't such command: %s", argv[2]);
}