From 1c39aeca008ced62dae8245f8d6f9963f4c97b47 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 12 Apr 2018 11:12:38 +0000
Subject: [PATCH] Try to fix
---
src/network.c | 13 ++++++++++---
1 files changed, 10 insertions(+), 3 deletions(-)
diff --git a/src/network.c b/src/network.c
index bfade3c..8c713fb 100644
--- a/src/network.c
+++ b/src/network.c
@@ -555,6 +555,9 @@
free(boxes);
free_ptrs((void **)probs, l.w*l.h*l.n);
+
+ //correct_region_boxes(dets, l.w*l.h*l.n, w, h, net_w, net_h, relative);
+ correct_yolo_boxes(dets, l.w*l.h*l.n, w, h, net_w, net_h, relative, letter);
}
void fill_network_boxes(network *net, int w, int h, float thresh, float hier, int *map, int relative, detection *dets, int letter)
@@ -757,7 +760,7 @@
layer *l = &net.layers[j];
if (l->type == CONVOLUTIONAL) {
- printf(" Fuse Convolutional layer \t\t l->size = %d \n", l->size);
+ //printf(" Merges Convolutional-%d and batch_norm \n", j);
if (l->batch_normalize) {
int f;
@@ -775,11 +778,15 @@
}
l->batch_normalize = 0;
- push_convolutional_layer(*l);
+#ifdef GPU
+ if (gpu_index >= 0) {
+ push_convolutional_layer(*l);
+ }
+#endif
}
}
else {
- printf(" Skip layer: %d \n", l->type);
+ //printf(" Fusion skip layer type: %d \n", l->type);
}
}
}
--
Gitblit v1.10.0