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

Optimizing network_predict_image to resize only if necessary.

Browse Source 
This speeds up the processing considerably if the user is so
kind to resize the image prior to doing detection.
This commit is contained in:
John Aughey
2019-02-05 11:30:38 -06:00
parent edfdf2c20e
commit 8726d7b0db
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/network.c
View File

@ -731,10 +731,17 @@ char *detection_to_json(detection *dets, int nboxes, int classes, char **names,
float *network_predict_image(network *net, image im)
{
//image imr = letterbox_image(im, net->w, net->h);
image imr = resize_image(im, net->w, net->h);
set_batch_network(net, 1);
float *p = network_predict(*net, imr.data);
free_image(imr);
float *p;
if (im.w == net->w && im.h == net->h) {
// Input image is the same size as our net, predict on that image
p = network_predict(*net, im.data);
}
else {
// Need to resize image to the desired size for the net
image imr = resize_image(im, net->w, net->h);
p = network_predict(*net, imr.data);
free_image(imr);
}
return p;
}