mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
Minor Python and C API improvement
This commit is contained in:
@ -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);
|
||||
|
Reference in New Issue
Block a user