From 57fa745be879e9fcbe007651bd190d3a40de3f38 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Fri, 4 Aug 2017 23:31:12 +0300 Subject: [PATCH] Fixed speed of SO/DLL --- src/yolo_v2_class.cpp | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/yolo_v2_class.cpp b/src/yolo_v2_class.cpp index e8be427e..813a24f2 100644 --- a/src/yolo_v2_class.cpp +++ b/src/yolo_v2_class.cpp @@ -183,7 +183,15 @@ YOLODLL_API std::vector Detector::detect(image_t img, float thresh) im.h = img.h; im.w = img.w; - image sized = resize_image(im, net.w, net.h); + image sized; + + if (net.w == im.w && net.h == im.h) { + sized = make_image(im.w, im.h, im.c); + memcpy(sized.data, im.data, im.w*im.h*im.c * sizeof(float)); + } + else + sized = resize_image(im, net.w, net.h); + layer l = net.layers[net.n - 1]; float *X = sized.data;