I have trained and tested the Caffe model for the MNIST dataset which produces 85 to 90% accuracy when tested on Desktop computers. I have generated a header file after solving bugs in code_gen.py script. When exported on CMSIS-NN for testing image on the microcontroller, I only got 3% accuracy.
After changes in the parameter.h file as mentioned below.
View all questions in Machine Learning forum