Added flag -show_imgs for Training, to show augmented images with bboxes

This commit is contained in:
AlexeyAB
2019-04-05 16:04:26 +03:00
parent f19d1440cb
commit d7a95aefb2
4 changed files with 25 additions and 15 deletions

View File

@ -706,6 +706,7 @@ typedef struct load_args {
int mini_batch;
int track;
int augment_speed;
int show_imgs;
float jitter;
int flip;
float angle;
@ -771,7 +772,7 @@ LIB_API detection *make_network_boxes(network *net, float thresh, int *num);
LIB_API void reset_rnn(network *net);
LIB_API float *network_predict_image(network *net, image im);
LIB_API float validate_detector_map(char *datacfg, char *cfgfile, char *weightfile, float thresh_calc_avg_iou, const float iou_thresh, const int map_points, network *existing_net);
LIB_API void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear, int dont_show, int calc_map, int mjpeg_port);
LIB_API void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear, int dont_show, int calc_map, int mjpeg_port, int show_imgs);
LIB_API void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh,
float hier_thresh, int dont_show, int ext_output, int save_labels, char *outfile);
LIB_API int network_width(network *net);