From baf3fcb491ee1a5e083fbdfdf2c28aaf36488e92 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 16 Feb 2018 23:26:08 +0000
Subject: [PATCH] Calculation TP, FP, FN, avg_IoU for default threshold during mAP calculation
---
src/compare.c | 5 ++---
1 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/compare.c b/src/compare.c
index 76e0b60..4fd266c 100644
--- a/src/compare.c
+++ b/src/compare.c
@@ -9,7 +9,6 @@
void train_compare(char *cfgfile, char *weightfile)
{
- data_seed = time(0);
srand(time(0));
float avg_loss = -1;
char *base = basecfg(cfgfile);
@@ -307,7 +306,7 @@
qsort(boxes, N, sizeof(sortable_bbox), elo_comparator);
N /= 2;
- for(round = 1; round <= 20; ++round){
+ for(round = 1; round <= 100; ++round){
clock_t round_time=clock();
printf("Round: %d\n", round);
@@ -316,7 +315,7 @@
bbox_fight(net, boxes+i*2, boxes+i*2+1, classes, class);
}
qsort(boxes, N, sizeof(sortable_bbox), elo_comparator);
- N = (N*9/10)/2*2;
+ if(round <= 20) N = (N*9/10)/2*2;
printf("Round: %f secs, %d remaining\n", sec(clock()-round_time), N);
}
--
Gitblit v1.10.0