aitrios-rpi-model-zoo

Please read the Site Policy of GitHub and understand the usage conditions.

This repository provides sample AI models and inference scripts for Raspberry Pi AI Camera Module.

Before running the Python demos, please make sure to install OpenCV:

sudo apt install python3-opencv python3-munkres

Python demo is in models folder.

The model (*.rpk), label, and config files are not included in this repository. These are output files from Brain Builder.

In anomaly-detection/brainbuilder

python app.py --model YOUR_MODEL.rpk --config neurala_hifi_ppl_parameters.json --fps 15

In classification/brainbuilder

python app.py --model YOUR_MODEL.rpk --labels labels.txt --fps 15

In object-detection/brainbuilder

python app.py --model YOUR_MODEL.rpk --labels labels.txt --fps 15

The model (*.rpk), label are not included in this repository. These are output files from aitrios-rpi-training-samples

In anomaly-detection/rd4ad

python app.py --model YOUR_MODEL.rpk --fps 15 --overlay

In classification/mobilenet_v2

python app.py

In object-detection/efficientdet_lite0_pp

python app.py

In segmentation/deeplabv3plus

python app.py

In pose_estimation/higherhrnet_coco

python app.py

In the table below you can find information regarding all models in the model zoo.
Please use the following models or datasets in accordance with the licenses of their respective sources.