Merge pull request #387 from openmv/fix_col_buf

NN: fix col buffer size.
This commit is contained in:
Ibrahim Abd Elkader 2018-10-03 19:20:02 +02:00 committed by GitHub
commit c8defe3b26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -361,7 +361,7 @@ int nn_run_network(nn_t *net, image_t *img, rectangle_t *roi, bool softmax)
q7_t *buffer1 = fb_alloc(net->max_scrbuf_size);
q7_t *buffer2 = buffer1 + net->max_layer_size;
q7_t *col_buffer = fb_alloc(net->max_colbuf_size);
q7_t *col_buffer = fb_alloc(net->max_colbuf_size * 2);
while (layer != NULL) {
layer_t *prev_layer = layer->prev;