Custom PyTorch ML block example for Edge Impulse
This repository is an example on how to bring a custom transfer learning model into Edge Impulse. This repository contains a small fully-connected model built in PyTorch. If you want to see a more complex PyTorch example, see edgeimpulse/yolov5. Or if you’re looking for the Keras version of this repository, see edgeimpulse/example-custom-ml-block-keras.
As a primer, read the Adding custom transfer learning models page in the Edge Impulse docs.
To test this locally:
-
Create a new Edge Impulse project, and add data from the continuous gestures dataset.
-
Under Create impulse add a ‘Spectral features’ processing block, and a random ML block.
-
Generate features for the DSP block.
-
Then go to