From 1479cefc6f536687a9e2005c55c81dc605e5eb96 Mon Sep 17 00:00:00 2001 From: iabdalkader Date: Wed, 3 Oct 2018 19:07:09 +0200 Subject: [PATCH] NN: fix col buffer size. * This is a workaround until a proper fix is found. --- src/omv/nn/nn.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/omv/nn/nn.c b/src/omv/nn/nn.c index 966cca874..d67bd576d 100644 --- a/src/omv/nn/nn.c +++ b/src/omv/nn/nn.c @@ -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;