mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Fixed speed of SO/DLL

Browse Source
This commit is contained in:
AlexeyAB
2017-08-04 23:31:12 +03:00
parent 52462f2ba6
commit 57fa745be8
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/yolo_v2_class.cpp
View File

@ -183,7 +183,15 @@ YOLODLL_API std::vector<bbox_t> 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;