From 8f1f5cbf8321b6b313d8f455d596290e7b8bb3f7 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sat, 24 Mar 2018 20:35:05 +0000
Subject: [PATCH] 3.5x times accelerated data augmentation using OpenCV for training Yolo on DGX and multi-GPU
---
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