► STM32 model zoo v4.0 (online)
Free AI models optimized for STM32
The STM32 AI model zoo is the largest collection of pre-trained AI models (140+) optimized to run on STM32 microcontrollers. Available on GitHub, it is a valuable resource for anyone looking to add AI capabilities to their STM32-based projects.
The model zoo also includes the support of the Neural-ART Accelerator NPU embedded in the STM32N6.
New in version 4.0
– Support for ST Edge AI Core v3.0
– New Keras 3 and Pytorch models
– Mixed precision quantization service & models
– New use-cases: AFCI, Re‑ID, Face Detection
Get the best edge AI model for your application
– A large collection of application-oriented models in PyTorch, TensorFlow, and ONNX, ready for re-training.
– Scripts to easily quantize and retrain any model from user datasets.
– Application code examples automatically generated from user AI model.
COMPUTER VISION
► Image Classification (IC) – 24 model families
► Object Detection (OD) – 12 model families (incl. ST_YOLO, YOLOv8 and YOLO11)
► Pose Estimation – 6 model families (incl. YOLOv8 and YOLO11)
► Semantic Segmentation – 1 model family
► Instance Segmentation – 2 model families (incl. YOLOv8 and YOLO11)
► Depth Estimation – 1 model family
► Neural Style Transfer – 1 model family
► Re‑Identification – 2 model families
► Face Detection – 2 model families
AUDIO
► Audio Event Detection (AED) – 3 model families
► Speech Enhancement – 1 model family
TIME SERIES
► Human Activity Recognition (HAR) – 2 model families
► Hand Posture Recognition (HPR) – 1 model family
► Arc Fault Detection (AFCI) – 2 model families
STM32 model zoo setup essentials
To set up the STM32 Model Zoo, you will need to create a myST account, access STM32Cube.AI via ST Edge AI Developer Cloud or local installation, ensure Python version 3.12.9 is installed, and if using a GPU, install the appropriate drivers along with CUDA and CUDNN, avoiding WSL on Windows for optimal GPU acceleration.
STM32 is now on Hugging Face
All STM32 model zoo services are available in a docker.
Browse NN models performances with the model cards.
Find application examples
based on the models of the zoo
Using AI opens the door to new application possibilities! Discover inspiring real-world examples and use our resources to create your own application.
Get started with the Model zoo
Discover how to optimize your AI Neural Network and create processing libraries for your STM32 project
