Problem
It is often difficult to be able to run inference on trained models of open source projects. This is mainly because most deep learning repos are all about describing new architectures, explaining training processes and/or publishing amazing metric results. The focus is more on the model description and explanation rather than on the model use.
When we want to run some model, we often encounter complex installation steps or even model unavailability (train it yourself!).
Goal
The goal of this repo is to provide a place where we can make use of the trained models. Each model will be hosted in it's own site where a README will guide the user over simple and easy steps on how to run the model (often involving an auxiliary class also provided in the site). All models will be also hosted on this repo, under the releases
section.
🙌
Contribute! The only way to grow this collection is with your help. If you know how to run a traditional model and/or you built one and you wish to share it, you're welcome. Read contribute.md.
Note
Python only for now!
How to use
- Pick a category.
- Pick the task
- Pick the model and go to the model site!
Categories:
This index will take you to all models of the category, regardless framework
Computer Vision tasks
- Image Classification
- Object Detection
- Pose Estimation
- Face Detection
- Instance Segmentation
- Image Enhancement
- Image Captioning
- Other
Natural Language Processing tasks
Audio tasks
Image Classification
- Inception trained on ImageNet: Classify entire images into 1000 classes, like "Zebra", "Panda", and "Dishwasher".
Object Detection
- SSD MobileNet trained on Coco: Locate and classify objects into 80 classes with high speed.
- FasterRCNN Resnet 50 trained Coco: Locate and classify objects into 80 classes with a high accuracy.
- YOLOv2 trained Coco: Locate and classify objects into 80 classes with high speed.
Pose Estimation
- OpenPose trained on COCO Keypoint: Find body keypoints (knees, arms, eyes, hip, etc.) in an image.
Instance Segmentation
- MaskRCNN Inception trained on COCO: Segment all objects intances given a color image.
Image Enhancement
- Colorful Image Colorization: Colorize black and white images!
Image Captioning
- im2txt: Caption an image. Implementation of Show And Tell
Other Computer Vision Models
- Fast Style Transfer: Add styles from famous paintings to any photo.
Speech To Text
- Deep Speech: Convert speech audio to text. Based on mozilla/DeepSpeech