Merge pull request #492 from openmv/fix_nn

Fix NN enum sizes.
This commit is contained in:
Ibrahim Abd Elkader 2019-03-21 18:59:50 +02:00 committed by GitHub
commit d15a1e35bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View File

@ -94,7 +94,7 @@ int nn_load_network(nn_t *net, const char *path)
layer_t *prev_layer = NULL;
for (int i=0; i<net->n_layers; i++) {
layer_t *layer;
layer_type_t layer_type;
uint32_t layer_type;
// Read layer type
read_data(&fp, &layer_type, 4);

View File

@ -27,7 +27,7 @@ typedef enum {
} network_type_t;
#define NN_LAYER_BASE \
layer_type_t type; \
uint32_t type; \
uint32_t n, c, h, w;\
struct _layer *prev;\
struct _layer *next \
@ -62,7 +62,7 @@ typedef struct {
typedef struct {
NN_LAYER_BASE;
pool_type_t ptype;
uint32_t ptype;
uint32_t krn_dim;
uint32_t krn_str;
uint32_t krn_pad;