Machine Learning Resources (Keep Updating)
Paper
Learning Resources
- General
- NLP
- CV
- CS231n: Convolutional Neural Networks for Visual Recognition
- Awesome-Computer-Vision
- MedicalTorch
- Built on top of PyTorch. Basic functionalities of the tools offered in MedicalTroch: pre-processing of images, transformations, and data loaders.
- The Computer Vision Industry
- Computer-Vision-Basics-with-Python-Keras-and-OpenCV
- Blogs
Database
- General
- NLP
- CV
- Classical:
- Object Detection:
- Object Classification:
- Key Points:
Pre-trained CV Models
Tensorflow
| Model Name | Description | Framework |
|---|---|---|
| ObjectDetection | Localizing and identifying multiple objects in a single image. | Tensorflow |
| Mask R-CNN | The model generates bounding boxes and segmentation masks for each instance of an object in the image. It’s based on Feature Pyramid Network (FPN) and a ResNet101 backbone. | Tensorflow |
| Faster-RCNN | This is an experimental Tensorflow implementation of Faster RCNN - a convnet for object detection with a region proposal network. | Tensorflow |
| YOLO TensorFlow | This is tensorflow implementation of the YOLO:Real-Time Object Detection. | Tensorflow |
| YOLO TensorFlow ++ | TensorFlow implementation of ‘YOLO: Real-Time Object Detection’, with training and an actual support for real-time running on mobile devices. | Tensorflow |
| MobileNet | MobileNets trade off between latency, size and accuracy while comparing favorably with popular models from the literature. | Tensorflow |
| DeepLab | Deep labeling for semantic image segmentation. | Tensorflow |
| Colornet | Neural Network to colorize grayscale images. | Tensorflow |
| SRGAN | Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network. | Tensorflow |
| DeepOSM | Train TensorFlow neural nets with OpenStreetMap features and satellite imagery. | Tensorflow |
| Domain Transfer Network | Implementation of Unsupervised Cross-Domain Image Generation. | Tensorflow |
| Show, Attend and Tell | Attention Based Image Caption Generator. | Tensorflow |
| android-yolo | Real-time object detection on Android using the YOLO network, powered by TensorFlow. | Tensorflow |
| DCSCN Super Resolution | This is a tensorflow implementation of “Fast and Accurate Image Super Resolution by Deep CNN with Skip Connection and Network in Network”, a deep learning based Single-Image Super-Resolution (SISR) model. | Tensorflow |
| GAN-CLS | This is an experimental tensorflow implementation of synthesizing images. | Tensorflow |
| U-Net | For Brain Tumor Segmentation. | Tensorflow |
| Improved CycleGAN | Unpaired Image to Image Translation. | Tensorflow |
| Im2txt | Image-to-text neural network for image captioning. | Tensorflow |
| Street | Identify the name of a street (in France) from an image using a Deep RNN. | Tensorflow |
| SLIM | Image classification models in TF-Slim. | Tensorflow |
| DELF | Deep local features for image matching and retrieval. | Tensorflow |
| Compression | Compressing and decompressing images using a pre-trained Residual GRU network. | Tensorflow |
| AttentionOCR | A model for real-world image text extraction. | Tensorflow |
Keras
| Model Name | Description | Framework |
|---|---|---|
| Mask R-CNN | The model generates bounding boxes and segmentation masks for each instance of an object in the image. It’s based on Feature Pyramid Network (FPN) and a ResNet101 backbone. | Keras |
| VGG16 | Very Deep Convolutional Networks for Large-Scale Image Recognition. | Keras |
| VGG19 | Very Deep Convolutional Networks for Large-Scale Image Recognition. | Keras |
| ResNet | Deep Residual Learning for Image Recognition. | Keras |
| Image analogies | Generate image analogies using neural matching and blending. | Keras |
| Popular Image Segmentation Models | Implementation of Segnet, FCN, UNet and other models in Keras. | Keras |
| Ultrasound nerve segmentation | This tutorial shows how to use Keras library to build deep neural network for ultrasound image nerve segmentation. | Keras |
| DeepMask object segmentation | This is a Keras-based Python implementation of DeepMask- a complex deep neural network for learning object segmentation masks. | Keras |
| Monolingual and Multilingual Image Captioning | This is the source code that accompanies Multilingual Image Description with Neural Sequence Models. | Keras |
| pix2pix | Keras implementation of Image-to-Image Translation with Conditional Adversarial Networks by Phillip Isola, Jun-Yan Zhu, Tinghui Zhou, Alexei A. | Keras |
| Colorful Image colorization | B&W to color. | Keras |
| CycleGAN | Implementation of Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks. | Keras |
| DualGAN | Implementation of DualGAN: Unsupervised Dual Learning for Image-to-Image Translation. | Keras |
| Super-Resolution GAN | Implementation of Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network. | Keras |
PyTorch
| Model Name | Description | Framework |
|---|---|---|
| FastPhotoStyle | A Closed-form Solution to Photorealistic Image Stylization. | PyTorch |
| pytorch-CycleGAN-and-pix2pix | A Closed-form Solution to Photorealistic Image Stylization. | PyTorch |
| maskrcnn-benchmark | Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch. | PyTorch |
| deep-image-prior | Image restoration with neural networks but without learning. | PyTorch |
| StarGAN | StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Translation. | PyTorch |
| faster-rcnn.pytorch | This project is a faster faster R-CNN implementation, aimed to accelerating the training of faster R-CNN object detection models. | PyTorch |
| pix2pixHD | Synthesizing and manipulating 2048x1024 images with conditional GANs. | PyTorch |
| Augmentor | Image augmentation library in Python for machine learning. | PyTorch |
| albumentations | Fast image augmentation library. | PyTorch |
| Deep Video Analytics | Deep Video Analytics is a platform for indexing and extracting information from videos and images | PyTorch |
| semantic-segmentation-pytorch | Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset. | PyTorch |
| An End-to-End Trainable Neural Network for Image-based Sequence Recognition | This software implements the Convolutional Recurrent Neural Network (CRNN), a combination of CNN, RNN and CTC loss for image-based sequence recognition tasks, such as scene text recognition and OCR. | PyTorch |
| UNIT | PyTorch Implementation of our Coupled VAE-GAN algorithm for Unsupervised Image-to-Image Translation. | PyTorch |
| Neural Sequence labeling model | Sequence labeling models are quite popular in many NLP tasks, such as Named Entity Recognition (NER), part-of-speech (POS) tagging and word segmentation. | PyTorch |
| faster rcnn | This is a PyTorch implementation of Faster RCNN. This project is mainly based on py-faster-rcnn and TFFRCNN. For details about R-CNN please refer to the paper Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks by Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun. | PyTorch |
| pytorch-semantic-segmentation | PyTorch for Semantic Segmentation. | PyTorch |
| EDSR-PyTorch | PyTorch version of the paper ‘Enhanced Deep Residual Networks for Single Image Super-Resolution’. | PyTorch |
| image-classification-mobile | Collection of classification models pretrained on the ImageNet-1K. | PyTorch |
| FaderNetworks | Fader Networks: Manipulating Images by Sliding Attributes - NIPS 2017. | PyTorch |
| neuraltalk2-pytorch | Image captioning model in pytorch (finetunable cnn in branch with_finetune). | PyTorch |
| RandWireNN | Implementation of: “Exploring Randomly Wired Neural Networks for Image Recognition”. | PyTorch |
| stackGAN-v2 | Pytorch implementation for reproducing StackGAN_v2 results in the paper StackGAN++. | PyTorch |
| Detectron models for Object Detection | This code allows to use some of the Detectron models for object detection from Facebook AI Research with PyTorch. | PyTorch |
| DEXTR-PyTorch | This paper explores the use of extreme points in an object (left-most, right-most, top, bottom pixels) as input to obtain precise object segmentation for images and videos. | PyTorch |
| pointnet.pytorch | Pytorch implementation for “PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation. | PyTorch |
| self-critical.pytorch | This repository includes the unofficial implementation Self-critical Sequence Training for Image Captioning and Bottom-Up and Top-Down Attention for Image Captioning and Visual Question Answering. | PyTorch |
| vnet.pytorch | A Pytorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation. | PyTorch |
| piwise | Pixel-wise segmentation on VOC2012 dataset using pytorch. | PyTorch |
| pspnet-pytorch | PyTorch implementation of PSPNet segmentation network. | PyTorch |
| pytorch-SRResNet | Pytorch implementation for Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network. | PyTorch |
| PNASNet.pytorch | PyTorch implementation of PNASNet-5 on ImageNet. | PyTorch |
| img_classification_pk_pytorch | Quickly comparing your image classification models with the state-of-the-art models. | PyTorch |
| Deep Neural Networks are Easily Fooled | High Confidence Predictions for Unrecognizable Images. | PyTorch |
| pix2pix-pytorch | PyTorch implementation of “Image-to-Image Translation Using Conditional Adversarial Networks”. | PyTorch |
| NVIDIA/semantic-segmentation | A PyTorch Implementation of Improving Semantic Segmentation via Video Propagation and Label Relaxation, In CVPR2019. | PyTorch |
| Neural-IMage-Assessment | A PyTorch Implementation of Neural IMage Assessment. | PyTorch |
| torchxrayvision | Pretrained models for chest X-ray (CXR) pathology predictions. Medical, Healthcare, Radiology | PyTorch |
Useful API
- General
