mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Use -dont_show -show_imgs for Training - saves augmented images without showing

Browse Source
This commit is contained in:
AlexeyAB
2019-04-06 15:13:47 +03:00
parent 97679224ba
commit 71d20264d1
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/data.c
View File

@ -856,8 +856,10 @@ data load_data_detection(int n, char **paths, int m, int w, int h, int c, int bo
} }
save_image(ai, buff); save_image(ai, buff);
show_image(ai, buff); if (show_imgs == 1) {
wait_until_press_key_cv(); show_image(ai, buff);
wait_until_press_key_cv();
}
printf("\nYou use flag -show_imgs, so will be saved aug_...jpg images. Click on window and press ESC button \n"); printf("\nYou use flag -show_imgs, so will be saved aug_...jpg images. Click on window and press ESC button \n");
} }
@ -951,8 +953,10 @@ data load_data_detection(int n, char **paths, int m, int w, int h, int c, int bo
} }
show_image(sized, buff); show_image(sized, buff);
save_image(sized, buff); if (show_imgs == 1) {
wait_until_press_key_cv(); save_image(sized, buff);
wait_until_press_key_cv();
}
printf("\nYou use flag -show_imgs, so will be saved aug_...jpg images. Press Enter: \n"); printf("\nYou use flag -show_imgs, so will be saved aug_...jpg images. Press Enter: \n");
getchar(); getchar();
} }

1
src/detector.c
View File

@ -127,6 +127,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
args.exposure = net.exposure; args.exposure = net.exposure;
args.saturation = net.saturation; args.saturation = net.saturation;
args.hue = net.hue; args.hue = net.hue;
if (dont_show && show_imgs) show_imgs = 2;
args.show_imgs = show_imgs; args.show_imgs = show_imgs;
#ifdef OPENCV #ifdef OPENCV