From 92e6e8eece3b789616430cad51a5afdb5a3153fc Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Sun, 15 Mar 2020 18:08:00 +0300 Subject: [PATCH] Minor fix --- src/detector.c | 2 +- src/yolo_layer.c | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/detector.c b/src/detector.c index 92ce5864..d403f38c 100644 --- a/src/detector.c +++ b/src/detector.c @@ -191,7 +191,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i if (dim_w < net.resize_step) dim_w = net.resize_step; if (dim_h < net.resize_step) dim_h = net.resize_step; int dim_b = (init_b * max_dim_w * max_dim_h) / (dim_w * dim_h); - int new_dim_b = (int)(dim_b * 0.9); + int new_dim_b = (int)(dim_b * 0.8); if (new_dim_b > init_b) dim_b = new_dim_b; args.w = dim_w; diff --git a/src/yolo_layer.c b/src/yolo_layer.c index 31fe7249..1ea6bf4d 100644 --- a/src/yolo_layer.c +++ b/src/yolo_layer.c @@ -541,6 +541,9 @@ void forward_yolo_layer(const layer l, network_state state) } } + if (count == 0) count = 1; + if (class_count == 0) class_count = 1; + //*(l.cost) = pow(mag_array(l.delta, l.outputs * l.batch), 2); //printf("Region %d Avg IOU: %f, Class: %f, Obj: %f, No Obj: %f, .5R: %f, .75R: %f, count: %d\n", state.index, avg_iou / count, avg_cat / class_count, avg_obj / count, avg_anyobj / (l.w*l.h*l.n*l.batch), recall / count, recall75 / count, count);