Minor Python and C API improvement

This commit is contained in:
AlexeyAB
2019-02-06 14:38:12 +03:00
parent b76f1c0006
commit c50b0e0c8a
9 changed files with 15 additions and 11 deletions

View File

@ -738,6 +738,7 @@ float *network_predict_image(network *net, image im)
{
//image imr = letterbox_image(im, net->w, net->h);
float *p;
if(net->batch != 1) set_batch_network(net, 1);
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);