Skip to content

Train a neural network and export weights as a string to copy into Arduino sketches

Notifications You must be signed in to change notification settings

JonComo/Arduino_weight_trainer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Arduino Weight Trainer (AWT)

Made for this project https://github.com/IdleHandsProject/makennbot Or any arduino project that uses the neural network written here http://robotics.hobbizine.com/arduinoann.html

You'll need the following packages (you can pip install each of them)

tensorflow

keras

numpy

jupyter notebook

matplotlib

To use:

Setup x_train and y_train in the notebook to be the examples you want the network to train on.

Run the cells sequentially and you'll see your network fit the training data.

The final cell prints out a nicely formatted text version of the trained weights that you can copy directly into your arduino sketch!

You have to disable weight randomization if you want to use your trained weights, comment out these lines in the sketch to do so:

HiddenWeights[j][i] = 2.0 * ( Rando - 0.5 ) * InitialWeightMax ; OutputWeights[j][i] = 2.0 * ( Rando - 0.5 ) * InitialWeightMax ;

That's about it, let me know how it works!

About

Train a neural network and export weights as a string to copy into Arduino sketches

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published