parent: c1bb8c12 | patch | commit | ignore whitespace
AlexeyAB
2018-05-19 7b7c669f4952479ac7667cf7def4d09109cc6789 
Added flag -save_labels to save detections as training labels txt-files (pseudo labeling concept)
2 files modified
97 ■■■■■ changed files
src/detector.c 85 ●●●●● patch | view | raw | blame | history
src/image.c 12 ●●●●● patch | view | raw | blame | history 
 src/detector.c


@@ -36,6 +36,17 @@




static int coco_ids[] = {1,2,3,4,5,6,7,8,9,10,11,13,14,15,16,17,18,19,20,21,22,23,24,25,27,28,31,32,33,34,35,36,37,38,39,40,41,42,43,44,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,67,70,72,73,74,75,76,77,78,79,80,81,82,84,85,86,87,88,89,90};




void replace_image_to_label(char *input_path, char *output_path) {


    find_replace(input_path, "images", "labels", output_path);


    find_replace(output_path, "JPEGImages", "labels", output_path);


    find_replace(output_path, ".jpg", ".txt", output_path);


    find_replace(output_path, ".png", ".txt", output_path);


    find_replace(output_path, ".bmp", ".txt", output_path);


    find_replace(output_path, ".JPG", ".txt", output_path);


    find_replace(output_path, ".JPEG", ".txt", output_path);


    find_replace(output_path, ".ppm", ".txt", output_path);


}




void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear, int dont_show)


{


    list *options = read_data_cfg(datacfg);


@@ -474,14 +485,15 @@


        if (nms) do_nms_obj(dets, nboxes, 1, nms);




        char labelpath[4096];


        find_replace(path, "images", "labels", labelpath);


        find_replace(labelpath, "JPEGImages", "labels", labelpath);


        find_replace(labelpath, ".jpg", ".txt", labelpath);


        find_replace(labelpath, ".png", ".txt", labelpath);


        find_replace(labelpath, ".bmp", ".txt", labelpath);


        find_replace(labelpath, ".JPG", ".txt", labelpath);


        find_replace(labelpath, ".JPEG", ".txt", labelpath);


        find_replace(labelpath, ".ppm", ".txt", labelpath);


        replace_image_to_label(path, labelpath);


        //find_replace(path, "images", "labels", labelpath);


        //find_replace(labelpath, "JPEGImages", "labels", labelpath);


        //find_replace(labelpath, ".jpg", ".txt", labelpath);


        //find_replace(labelpath, ".png", ".txt", labelpath);


        //find_replace(labelpath, ".bmp", ".txt", labelpath);


        //find_replace(labelpath, ".JPG", ".txt", labelpath);


        //find_replace(labelpath, ".JPEG", ".txt", labelpath);


        //find_replace(labelpath, ".ppm", ".txt", labelpath);




        int num_labels = 0;


        box_label *truth = read_boxes(labelpath, &num_labels);


@@ -632,14 +644,15 @@


            if (nms) do_nms_sort(dets, nboxes, l.classes, nms);




            char labelpath[4096];


            find_replace(path, "images", "labels", labelpath);


            find_replace(labelpath, "JPEGImages", "labels", labelpath);


            find_replace(labelpath, ".jpg", ".txt", labelpath);


            find_replace(labelpath, ".png", ".txt", labelpath);


            find_replace(labelpath, ".bmp", ".txt", labelpath);


            find_replace(labelpath, ".JPG", ".txt", labelpath);


            find_replace(labelpath, ".JPEG", ".txt", labelpath);


            find_replace(labelpath, ".ppm", ".txt", labelpath);


            replace_image_to_label(path, labelpath);


            //find_replace(path, "images", "labels", labelpath);


            //find_replace(labelpath, "JPEGImages", "labels", labelpath);


            //find_replace(labelpath, ".jpg", ".txt", labelpath);


            //find_replace(labelpath, ".png", ".txt", labelpath);


            //find_replace(labelpath, ".bmp", ".txt", labelpath);


            //find_replace(labelpath, ".JPG", ".txt", labelpath);


            //find_replace(labelpath, ".JPEG", ".txt", labelpath);


            //find_replace(labelpath, ".ppm", ".txt", labelpath);


            int num_labels = 0;


            box_label *truth = read_boxes(labelpath, &num_labels);


            int i, j;


@@ -896,14 +909,15 @@


    for (i = 0; i < number_of_images; ++i) {


        char *path = paths[i];


        char labelpath[4096];


        find_replace(path, "images", "labels", labelpath);


        find_replace(labelpath, "JPEGImages", "labels", labelpath);


        find_replace(labelpath, ".jpg", ".txt", labelpath);


        find_replace(labelpath, ".png", ".txt", labelpath);


        find_replace(labelpath, ".bmp", ".txt", labelpath);


        find_replace(labelpath, ".JPG", ".txt", labelpath);


        find_replace(labelpath, ".JPEG", ".txt", labelpath);


        find_replace(labelpath, ".ppm", ".txt", labelpath);


        replace_image_to_label(path, labelpath);


        //find_replace(path, "images", "labels", labelpath);


        //find_replace(labelpath, "JPEGImages", "labels", labelpath);


        //find_replace(labelpath, ".jpg", ".txt", labelpath);


        //find_replace(labelpath, ".png", ".txt", labelpath);


        //find_replace(labelpath, ".bmp", ".txt", labelpath);


        //find_replace(labelpath, ".JPG", ".txt", labelpath);


        //find_replace(labelpath, ".JPEG", ".txt", labelpath);


        //find_replace(labelpath, ".ppm", ".txt", labelpath);


        int num_labels = 0;


        box_label *truth = read_boxes(labelpath, &num_labels);


        //printf(" new path: %s \n", labelpath);


@@ -1056,7 +1070,7 @@


#endif // OPENCV




void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh,


                   float hier_thresh, int dont_show, int ext_output)


                   float hier_thresh, int dont_show, int ext_output, int save_labels)


{


    list *options = read_data_cfg(datacfg);


    char *name_list = option_find_str(options, "names", "data/names.list");


@@ -1113,16 +1127,18 @@


        if (!dont_show) {


            show_image(im, "predictions");


        }


/*




        // pseudo labeling concept - fast.ai


        if(save_labels)


        {


            char labelpath[4096];


            find_replace(input, ".jpg", ".txt", labelpath);


            find_replace(labelpath, ".png", ".txt", labelpath);


            find_replace(labelpath, ".bmp", ".txt", labelpath);


            find_replace(labelpath, ".JPG", ".txt", labelpath);


            find_replace(labelpath, ".JPEG", ".txt", labelpath);


            find_replace(labelpath, ".ppm", ".txt", labelpath);


            replace_image_to_label(input, labelpath);


            //find_replace(input, ".jpg", ".txt", labelpath);


            //find_replace(labelpath, ".png", ".txt", labelpath);


            //find_replace(labelpath, ".bmp", ".txt", labelpath);


            //find_replace(labelpath, ".JPG", ".txt", labelpath);


            //find_replace(labelpath, ".JPEG", ".txt", labelpath);


            //find_replace(labelpath, ".ppm", ".txt", labelpath);




            FILE* fw = fopen(labelpath, "wb");


            int i;


@@ -1143,7 +1159,7 @@


            }


            fclose(fw);


        }


*/




        free_detections(dets, nboxes);


        free_image(im);


        free_image(sized);


@@ -1193,6 +1209,7 @@


    // extended output in test mode (output of rect bound coords)


    // and for recall mode (extended output table-like format with results for best_class fit)


    int ext_output = find_arg(argc, argv, "-ext_output");


    int save_labels = find_arg(argc, argv, "-save_labels");


    if(argc < 4){


        fprintf(stderr, "usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]);


        return;


@@ -1229,7 +1246,7 @@


        if(strlen(weights) > 0)


            if (weights[strlen(weights) - 1] == 0x0d) weights[strlen(weights) - 1] = 0;


    char *filename = (argc > 6) ? argv[6]: 0;


    if(0==strcmp(argv[2], "test")) test_detector(datacfg, cfg, weights, filename, thresh, hier_thresh, dont_show, ext_output);


    if(0==strcmp(argv[2], "test")) test_detector(datacfg, cfg, weights, filename, thresh, hier_thresh, dont_show, ext_output, save_labels);


    else if(0==strcmp(argv[2], "train")) train_detector(datacfg, cfg, weights, gpus, ngpus, clear, dont_show);


    else if(0==strcmp(argv[2], "valid")) validate_detector(datacfg, cfg, weights, outfile);


    else if(0==strcmp(argv[2], "recall")) validate_detector_recall(datacfg, cfg, weights);




 src/image.c


@@ -441,6 +441,8 @@


{


    int i, j;


    if (!show_img) return;


    static int frame_id = 0;


    frame_id++;




    for (i = 0; i < num; ++i) {


        char labelstr[4096] = { 0 };


@@ -516,6 +518,16 @@


            color.val[1] = green * 256;


            color.val[2] = blue * 256;




            // you should create directory: result_img


            //static int img_id = 0;


            //img_id++;


            //char image_name[1024];


            //sprintf(image_name, "result_img/img_%d_%d_%d.jpg", frame_id, img_id, class_id);


            //CvRect rect = cvRect(pt1.x, pt1.y, pt2.x - pt1.x, pt2.y - pt1.y);


            //cvSetImageROI(show_img, rect);


            //cvSaveImage(image_name, show_img, 0);


            //cvResetImageROI(show_img);




            cvRectangle(show_img, pt1, pt2, color, width, 8, 0);


            //printf("left=%d, right=%d, top=%d, bottom=%d, obj_id=%d, obj=%s \n", left, right, top, bot, class_id, names[class_id]);


            cvRectangle(show_img, pt_text_bg1, pt_text_bg2, color, width, 8, 0);