From 537d135feba179636e9bbfe296e078d51f59914c Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 19 Mar 2018 23:16:51 +0000
Subject: [PATCH] Improve training performance - batch-norm using cuDNN.
---
src/detector.c | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/src/detector.c b/src/detector.c
index 9f608a0..3dfbce6 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -91,7 +91,7 @@
args.small_object = l.small_object;
args.d = &buffer;
args.type = DETECTION_DATA;
- args.threads = 8; // 64
+ args.threads = 64; // 8
args.angle = net.angle;
args.exposure = net.exposure;
@@ -1031,6 +1031,7 @@
}
image im = load_image_color(input,0,0);
image sized = resize_image(im, net.w, net.h);
+ //image sized = letterbox_image(im, net.w, net.h);
layer l = net.layers[net.n-1];
box *boxes = calloc(l.w*l.h*l.n, sizeof(box));
--
Gitblit v1.10.0