Fixed issues 8510 and 8511 (#8516)

* Fixed issues 8510 and 8511

* Update yolo.c

Return fprintf
This commit is contained in:
Sergey Nuzhny 2022-05-17 01:57:30 +03:00 committed by GitHub
parent 8a0bf84c19
commit 695d562918
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 5 deletions

View File

@ -517,7 +517,7 @@ layer parse_yolo(list *options, size_params params)
l.random = option_find_float_quiet(options, "random", 0);
l.track_history_size = option_find_int_quiet(options, "track_history_size", 5);
l.sim_thresh = option_find_int_quiet(options, "sim_thresh", 0.8);
l.sim_thresh = option_find_float_quiet(options, "sim_thresh", 0.8);
l.dets_for_track = option_find_int_quiet(options, "dets_for_track", 1);
l.dets_for_show = option_find_int_quiet(options, "dets_for_show", 1);
l.track_ciou_norm = option_find_float_quiet(options, "track_ciou_norm", 0.01);

View File

@ -190,7 +190,6 @@ void validate_yolo(char *cfgfile, char *weightfile)
}
}
if (fps) free(fps);
if (val) free(val);
if (val_resized) free(val_resized);
if (buf) free(buf);
@ -198,10 +197,12 @@ void validate_yolo(char *cfgfile, char *weightfile)
if (thr) free(thr);
fprintf(stderr, "Total Detection Time: %f Seconds\n", (double)(time(0) - start));
for(j = 0; j < classes; ++j){
fclose(fps[j]);
if (fps) {
for(j = 0; j < classes; ++j){
fclose(fps[j]);
}
free(fps);
}
free(fps);
}
void validate_yolo_recall(char *cfgfile, char *weightfile)