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

Merge pull request #2923 from willbattel/save-best-map

Browse Source 
Save weights with best mAP during training
This commit is contained in:
Alexey
2019-05-27 19:02:49 +03:00
committed by GitHub
parent 3aca0b7166 4fe2c997d1
commit 2b7fea761f
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/detector.c
View File

@ -110,6 +110,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
iter_save_last = get_current_batch(net); iter_save_last = get_current_batch(net);
iter_map = get_current_batch(net); iter_map = get_current_batch(net);
float mean_average_precision = -1; float mean_average_precision = -1;
float best_map = mean_average_precision;
load_args args = { 0 }; load_args args = { 0 };
args.w = net.w; args.w = net.w;
@ -241,7 +242,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
next_map_calc = fmax(next_map_calc, 400); next_map_calc = fmax(next_map_calc, 400);
if (calc_map) { if (calc_map) {
printf("\n (next mAP calculation at %d iterations) ", next_map_calc); printf("\n (next mAP calculation at %d iterations) ", next_map_calc);
if (mean_average_precision > 0) printf("\n Last accuracy mAP@0.5 = %2.2f %% ", mean_average_precision * 100); if (mean_average_precision > 0) printf("\n Last accuracy mAP@0.5 = %2.2f %%, best = %2.2f %% ", mean_average_precision * 100, best_map * 100);
} }
if (net.cudnn_half) { if (net.cudnn_half) {
@ -275,6 +276,14 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
iter_map = i; iter_map = i;
mean_average_precision = validate_detector_map(datacfg, cfgfile, weightfile, 0.25, 0.5, 0, &net_map);// &net_combined); mean_average_precision = validate_detector_map(datacfg, cfgfile, weightfile, 0.25, 0.5, 0, &net_map);// &net_combined);
printf("\n mean_average_precision (mAP@0.5) = %f \n", mean_average_precision); printf("\n mean_average_precision (mAP@0.5) = %f \n", mean_average_precision);
if (mean_average_precision > best_map) {
best_map = mean_average_precision;
printf("New best mAP!\n");
char buff[256];
sprintf(buff, "%s/%s_best.weights", backup_directory, base);
save_weights(net, buff);
}
draw_precision = 1; draw_precision = 1;
} }
#ifdef OPENCV #ifdef OPENCV