Directory of Pretrained AI


One of the best tips for building a AI is to not start from scratch, but to start from a pretrained AI that has been trained on a similar task. Below I'd like to share a list of various pretrained AI.

This blog post was inspired by this HackerNews question. Thank you nerfhammer!

If anyone knows of any notable additions, please comment below.

Terminology

Model Zoo - A collection of pretrained models.

Tensorflow

Link # of Models Library(s) Short description
Tensorflow's Model Zoo 30+ Tensorflow The official models are a collection of example models that use TensorFlow's high-level APIs.
Tensorflow's Detection Model Zoo 12 Tensorflow We provide a collection of detection models pre-trained on the COCO dataset, the Kitti dataset, and the Open Images dataset.
YOLO Pretrained Models 10 Tensorflow You only look once (YOLO) is a state-of-the-art, real-time object detection system.
Generative Model Collection 10 Tensorflow, PyTorch Tensorflow implementation of various GANs and VAEs. PyTorch Version is now available.
albertomontesg's Model Zoo 2 Keras Repository to share all the models that the community has found and worked with the Keras framework.
Trained Image Classification Models 5 Keras This repository contains code for the following Keras models: VGG16, VGG19, ResNet50, Inception v3, CRNN for music tagging.
The List of Pretrained Word Embeddings 6 Keras Models include Word2Vec, GloVe, fastText, dependency-based word embeddings, meta-embeddings, and LexVec.
Caffe's Model Zoo 40+ Caffe These models are learned and applied for problems ranging from simple regression, to large-scale visual classification, to Siamese networks for image similarity, to speech and robotics applications.
ResNet Models 3 Caffe These models are those used in ILSVRC and COCO 2015 competitions, which won the 1st places in: ImageNet classification, ImageNet detection, ImageNet localization, COCO detection, and COCO segmentation.
Awesome Torch 26 Torch A curated list of awesome Torch tutorials, projects and communities.
Torch's Model Zoo 18 Torch Publicly available models for all to use (check the license at which they are released). This is a community-maintained list.
Singa Model Zoo 4 Apache Singa Includes models for Cifar-10, Char-RNN, AlexNet, and GoogleNet.

Useful Tools

Link Short description
Caffe to Tensorflow Converter Convert Caffe models to TensorFlow. There are some minor bugs, but it is otherwise quite solid.
Caffee to Torch Converter Load Caffe networks in Torch7.