About this project
The ONNX Model Zoo is a curated repository of pre-trained, state-of-the-art machine learning models exported into the ONNX (Open Neural Network Exchange) format. ONNX is an open standard that defines a common set of operators and a common file format, enabling AI developers to use models across a variety of frameworks, tools, runtimes, and compilers.
The models in this repository are sourced from prominent open-source projects including timm (PyTorch Image Models), torchvision, torch_hub, and Hugging Face Transformers, and are exported using the open-source TurnkeyML toolchain. The collection is organized into several major categories:
Computer Vision models include image classification (MobileNet, ResNet, SqueezeNet, VGG, AlexNet, GoogleNet, CaffeNet, DenseNet, Inception variants, ShuffleNet, ZFNet, EfficientNet-Lite4, and MNIST digit recognition), object detection and image segmentation (Tiny YOLOv2, SSD, SSD-MobileNetV1, Faster-RCNN, Mask-RCNN, RetinaNet, YOLO v2/v3/v4, DUC, and FCN), body/face/gesture analysis (ArcFace, UltraFace, Emotion FerPlus, and age/gender classification), and image manipulation (super-resolution with sub-pixel CNN and fast neural style transfer).
Natural Language Processing models cover machine comprehension (Bidirectional Attention Flow, BERT-Squad, RoBERTa, GPT-2, and T5), machine translation (Neural Machine Translation and Google's NMT system), and language modeling (Deep Neural Network Language Models).
Additional categories include speech and audio processing (speech recognition with deep RNNs, deep voice text-to-speech, and WaveNet sound generation), visual question answering and dialog, and graph machine learning.
The repository uses Git LFS (Large File Storage) to handle large ONNX model files. INT8 quantized models are generated using Intel Neural Compressor, an open-source Python library that supports automatic accuracy-driven tuning strategies for quantization. Many models include Hugging Face Spaces for interactive demos.
Important note: The ONNX Model Zoo is being deprecated. As of July 1st, 2025, models will no longer be available for LFS download from this repository. The models have been migrated to Hugging Face at https://huggingface.co/onnxmodelzoo, where the community can continue to access and share them.
Comments
0 Rating appears after 10 ratings
Sign in to join the discussion.