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. |