mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Added flag -save_labels to save detections as training labels txt-files (pseudo labeling concept)
This commit is contained in:
@ -36,6 +36,17 @@ void draw_train_loss(IplImage* img, int img_size, float avg_loss, float max_img_
|
|||||||
|
|
||||||
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};
|
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)
|
void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear, int dont_show)
|
||||||
{
|
{
|
||||||
list *options = read_data_cfg(datacfg);
|
list *options = read_data_cfg(datacfg);
|
||||||
@ -474,14 +485,15 @@ void validate_detector_recall(char *datacfg, char *cfgfile, char *weightfile)
|
|||||||
if (nms) do_nms_obj(dets, nboxes, 1, nms);
|
if (nms) do_nms_obj(dets, nboxes, 1, nms);
|
||||||
|
|
||||||
char labelpath[4096];
|
char labelpath[4096];
|
||||||
find_replace(path, "images", "labels", labelpath);
|
replace_image_to_label(path, labelpath);
|
||||||
find_replace(labelpath, "JPEGImages", "labels", labelpath);
|
//find_replace(path, "images", "labels", labelpath);
|
||||||
find_replace(labelpath, ".jpg", ".txt", labelpath);
|
//find_replace(labelpath, "JPEGImages", "labels", labelpath);
|
||||||
find_replace(labelpath, ".png", ".txt", labelpath);
|
//find_replace(labelpath, ".jpg", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".bmp", ".txt", labelpath);
|
//find_replace(labelpath, ".png", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPG", ".txt", labelpath);
|
//find_replace(labelpath, ".bmp", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
//find_replace(labelpath, ".JPG", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".ppm", ".txt", labelpath);
|
//find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
||||||
|
//find_replace(labelpath, ".ppm", ".txt", labelpath);
|
||||||
|
|
||||||
int num_labels = 0;
|
int num_labels = 0;
|
||||||
box_label *truth = read_boxes(labelpath, &num_labels);
|
box_label *truth = read_boxes(labelpath, &num_labels);
|
||||||
@ -632,14 +644,15 @@ void validate_detector_map(char *datacfg, char *cfgfile, char *weightfile, float
|
|||||||
if (nms) do_nms_sort(dets, nboxes, l.classes, nms);
|
if (nms) do_nms_sort(dets, nboxes, l.classes, nms);
|
||||||
|
|
||||||
char labelpath[4096];
|
char labelpath[4096];
|
||||||
find_replace(path, "images", "labels", labelpath);
|
replace_image_to_label(path, labelpath);
|
||||||
find_replace(labelpath, "JPEGImages", "labels", labelpath);
|
//find_replace(path, "images", "labels", labelpath);
|
||||||
find_replace(labelpath, ".jpg", ".txt", labelpath);
|
//find_replace(labelpath, "JPEGImages", "labels", labelpath);
|
||||||
find_replace(labelpath, ".png", ".txt", labelpath);
|
//find_replace(labelpath, ".jpg", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".bmp", ".txt", labelpath);
|
//find_replace(labelpath, ".png", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPG", ".txt", labelpath);
|
//find_replace(labelpath, ".bmp", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
//find_replace(labelpath, ".JPG", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".ppm", ".txt", labelpath);
|
//find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
||||||
|
//find_replace(labelpath, ".ppm", ".txt", labelpath);
|
||||||
int num_labels = 0;
|
int num_labels = 0;
|
||||||
box_label *truth = read_boxes(labelpath, &num_labels);
|
box_label *truth = read_boxes(labelpath, &num_labels);
|
||||||
int i, j;
|
int i, j;
|
||||||
@ -896,14 +909,15 @@ void calc_anchors(char *datacfg, int num_of_clusters, int width, int height, int
|
|||||||
for (i = 0; i < number_of_images; ++i) {
|
for (i = 0; i < number_of_images; ++i) {
|
||||||
char *path = paths[i];
|
char *path = paths[i];
|
||||||
char labelpath[4096];
|
char labelpath[4096];
|
||||||
find_replace(path, "images", "labels", labelpath);
|
replace_image_to_label(path, labelpath);
|
||||||
find_replace(labelpath, "JPEGImages", "labels", labelpath);
|
//find_replace(path, "images", "labels", labelpath);
|
||||||
find_replace(labelpath, ".jpg", ".txt", labelpath);
|
//find_replace(labelpath, "JPEGImages", "labels", labelpath);
|
||||||
find_replace(labelpath, ".png", ".txt", labelpath);
|
//find_replace(labelpath, ".jpg", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".bmp", ".txt", labelpath);
|
//find_replace(labelpath, ".png", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPG", ".txt", labelpath);
|
//find_replace(labelpath, ".bmp", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
//find_replace(labelpath, ".JPG", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".ppm", ".txt", labelpath);
|
//find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
||||||
|
//find_replace(labelpath, ".ppm", ".txt", labelpath);
|
||||||
int num_labels = 0;
|
int num_labels = 0;
|
||||||
box_label *truth = read_boxes(labelpath, &num_labels);
|
box_label *truth = read_boxes(labelpath, &num_labels);
|
||||||
//printf(" new path: %s \n", labelpath);
|
//printf(" new path: %s \n", labelpath);
|
||||||
@ -1056,7 +1070,7 @@ void calc_anchors(char *datacfg, int num_of_clusters, int width, int height, int
|
|||||||
#endif // OPENCV
|
#endif // OPENCV
|
||||||
|
|
||||||
void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh,
|
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);
|
list *options = read_data_cfg(datacfg);
|
||||||
char *name_list = option_find_str(options, "names", "data/names.list");
|
char *name_list = option_find_str(options, "names", "data/names.list");
|
||||||
@ -1113,16 +1127,18 @@ void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filenam
|
|||||||
if (!dont_show) {
|
if (!dont_show) {
|
||||||
show_image(im, "predictions");
|
show_image(im, "predictions");
|
||||||
}
|
}
|
||||||
/*
|
|
||||||
// pseudo labeling concept - fast.ai
|
// pseudo labeling concept - fast.ai
|
||||||
|
if(save_labels)
|
||||||
{
|
{
|
||||||
char labelpath[4096];
|
char labelpath[4096];
|
||||||
find_replace(input, ".jpg", ".txt", labelpath);
|
replace_image_to_label(input, labelpath);
|
||||||
find_replace(labelpath, ".png", ".txt", labelpath);
|
//find_replace(input, ".jpg", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".bmp", ".txt", labelpath);
|
//find_replace(labelpath, ".png", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPG", ".txt", labelpath);
|
//find_replace(labelpath, ".bmp", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
//find_replace(labelpath, ".JPG", ".txt", labelpath);
|
||||||
find_replace(labelpath, ".ppm", ".txt", labelpath);
|
//find_replace(labelpath, ".JPEG", ".txt", labelpath);
|
||||||
|
//find_replace(labelpath, ".ppm", ".txt", labelpath);
|
||||||
|
|
||||||
FILE* fw = fopen(labelpath, "wb");
|
FILE* fw = fopen(labelpath, "wb");
|
||||||
int i;
|
int i;
|
||||||
@ -1143,7 +1159,7 @@ void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filenam
|
|||||||
}
|
}
|
||||||
fclose(fw);
|
fclose(fw);
|
||||||
}
|
}
|
||||||
*/
|
|
||||||
free_detections(dets, nboxes);
|
free_detections(dets, nboxes);
|
||||||
free_image(im);
|
free_image(im);
|
||||||
free_image(sized);
|
free_image(sized);
|
||||||
@ -1193,6 +1209,7 @@ void run_detector(int argc, char **argv)
|
|||||||
// extended output in test mode (output of rect bound coords)
|
// 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)
|
// 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 ext_output = find_arg(argc, argv, "-ext_output");
|
||||||
|
int save_labels = find_arg(argc, argv, "-save_labels");
|
||||||
if(argc < 4){
|
if(argc < 4){
|
||||||
fprintf(stderr, "usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]);
|
fprintf(stderr, "usage: %s %s [train/test/valid] [cfg] [weights (optional)]\n", argv[0], argv[1]);
|
||||||
return;
|
return;
|
||||||
@ -1229,7 +1246,7 @@ void run_detector(int argc, char **argv)
|
|||||||
if(strlen(weights) > 0)
|
if(strlen(weights) > 0)
|
||||||
if (weights[strlen(weights) - 1] == 0x0d) weights[strlen(weights) - 1] = 0;
|
if (weights[strlen(weights) - 1] == 0x0d) weights[strlen(weights) - 1] = 0;
|
||||||
char *filename = (argc > 6) ? argv[6]: 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], "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], "valid")) validate_detector(datacfg, cfg, weights, outfile);
|
||||||
else if(0==strcmp(argv[2], "recall")) validate_detector_recall(datacfg, cfg, weights);
|
else if(0==strcmp(argv[2], "recall")) validate_detector_recall(datacfg, cfg, weights);
|
||||||
|
12
src/image.c
12
src/image.c
@ -441,6 +441,8 @@ void draw_detections_cv_v3(IplImage* show_img, detection *dets, int num, float t
|
|||||||
{
|
{
|
||||||
int i, j;
|
int i, j;
|
||||||
if (!show_img) return;
|
if (!show_img) return;
|
||||||
|
static int frame_id = 0;
|
||||||
|
frame_id++;
|
||||||
|
|
||||||
for (i = 0; i < num; ++i) {
|
for (i = 0; i < num; ++i) {
|
||||||
char labelstr[4096] = { 0 };
|
char labelstr[4096] = { 0 };
|
||||||
@ -516,6 +518,16 @@ void draw_detections_cv_v3(IplImage* show_img, detection *dets, int num, float t
|
|||||||
color.val[1] = green * 256;
|
color.val[1] = green * 256;
|
||||||
color.val[2] = blue * 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);
|
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]);
|
//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);
|
cvRectangle(show_img, pt_text_bg1, pt_text_bg2, color, width, 8, 0);
|
||||||
|
Reference in New Issue
Block a user