From 9ba0beb413d8b443376c36c1701a0dbce1e8d566 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Thu, 17 May 2018 17:02:42 +0300 Subject: [PATCH] Fixed resize of network for low resolution network like 128x128 --- src/detector.c | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/detector.c b/src/detector.c index 5ccb6c53..b7fbfbd9 100644 --- a/src/detector.c +++ b/src/detector.c @@ -132,6 +132,8 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i int random_val = rand() % 12; int dim_w = (random_val + (init_w / 32 - 5)) * 32; // +-160 int dim_h = (random_val + (init_h / 32 - 5)) * 32; // +-160 + if (dim_w < 32) dim_w = 32; + if (dim_h < 32) dim_h = 32; printf("%d x %d \n", dim_w, dim_h); args.w = dim_w;