Fixed resize of network for low resolution network like 128x128

pull/851/head
AlexeyAB 7 years ago
parent e39d101109
commit 9ba0beb413
  1. 2
      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 random_val = rand() % 12;
int dim_w = (random_val + (init_w / 32 - 5)) * 32; // +-160 int dim_w = (random_val + (init_w / 32 - 5)) * 32; // +-160
int dim_h = (random_val + (init_h / 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); printf("%d x %d \n", dim_w, dim_h);
args.w = dim_w; args.w = dim_w;

Loading…
Cancel
Save