mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
NN: Fix output size.
This commit is contained in:
parent
8ce21ddb72
commit
1c0c8d744b
@ -257,10 +257,12 @@ int nn_load_network(nn_t *net, const char *path)
|
||||
|
||||
uint32_t layer_size = layer->c * layer->h * layer->w;
|
||||
net->max_layer_size = IM_MAX(net->max_layer_size, layer_size);
|
||||
if (layer->next == NULL) {
|
||||
net->output_size = layer->c;
|
||||
}
|
||||
layer = layer->next;
|
||||
}
|
||||
|
||||
net->output_size = net->layers[net->n_layers-1].c;
|
||||
printf("Max layer: %lu Max col buf: %lu Max scratch buf: %lu Output size:%lu\n\n",
|
||||
net->max_layer_size, net->max_colbuf_size, net->max_scrbuf_size, net->output_size);
|
||||
error:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user