mirror of
https://github.com/RIOT-OS/RIOT.git
synced 2024-12-29 04:50:03 +01:00
32 lines
963 B
Markdown
32 lines
963 B
Markdown
## ARM CMSIS-NN package
|
|
|
|
This application shows how to use the neural network API provided by the ARM CMSIS
|
|
package in order to determine the type of "object" present in an RGB image.
|
|
The image are part of the [SIFAR10 dataset](http://www.cs.toronto.edu/~kriz/cifar.html)
|
|
which contains 10 classes of objects: plane, car, cat, bird, deer, dog, frog,
|
|
horse, ship and truck.
|
|
|
|
Expected output
|
|
---------------
|
|
|
|
```
|
|
Predicted class: cat
|
|
```
|
|
|
|
Change the input image
|
|
----------------------
|
|
|
|
Use the `generate_image.py` script and the `-i` option to generate a new
|
|
input image.
|
|
For example, the following command
|
|
```
|
|
./generate_image.py -i 1
|
|
```
|
|
will generate an input containing an image with a boat.
|
|
|
|
The generated image is displayed at the end of the script execution, for visual
|
|
validation of the prediction made by the neural network running on the device.
|
|
|
|
Note that each time a new image is generated, the firmware must be rebuilt so
|
|
that it embeds the new image.
|