Top 125 pretrained-models open source projects

A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)

Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet

BERT for Multitask Learning

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

[CVPR2020] f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation https://arxiv.org/abs/2001.10331

ImageNet pre-trained models with batch normalization for the Caffe framework

Implementation of various human pose estimation models in pytorch on multiple datasets (MPII & COCO) along with pretrained models

End-to-End recipes for pre-training and fine-tuning BERT using Azure Machine Learning Service

[CVPR 2021] VirTex: Learning Visual Representations from Textual Annotations

Segmentation models with pretrained backbones. PyTorch.

An extension of Open3D to address 3D Machine Learning tasks

Spleeter is Deezer source separation library with pretrained models written in Python and uses Tensorflow. It makes it easy to train source separation model (assuming you have a dataset of isolated sources), and provides already trained state of the art model for performing various flavour of separation :

74.3% MobileNetV3-Large and 67.2% MobileNetV3-Small model on ImageNet

InceptionV3, InceptionV4, Inception-Resnet pretrained models for Torch7 and PyTorch

An unofficial Gluon FR Toolkit for face recognition. https://gluon-face.readthedocs.io

SQuAD Question Answering Using BERT, PyTorch

CV backbones including GhostNet, TinyNet and TNT, developed by Huawei Noah's Ark Lab.

A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

PyTorch-style and human-readable RegNet with a spectrum of pre-trained models

Detect various irregularities of a product as it moves along a conveyor belt.

HugsVision is a easy to use huggingface wrapper for state-of-the-art computer vision

Neural syntax annotator, supporting sequence labeling, lemmatization, and dependency parsing.

PyTorch implementation of "Pyramid Scene Parsing Network".

Finetuning any DNN for better embedding on neural search tasks

A collection of pre-trained RL agents using Stable Baselines3

Multi-Task Pre-Training for Plug-and-Play Task-Oriented Dialogue System (ACL 2022)

Code for our paper CPPE - 5 (Medical Personal Protective Equipment), a new challenging object detection dataset

No description or website provided.

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation

Deep Learning Transformer models in MATLAB

PiGallery: AI-powered Self-hosted Secure Multi-user Image Gallery and Detailed Image analysis using Machine Learning, EXIF Parsing and Geo Tagging

Tensorflow2.3的文本分类项目，支持各种分类模型，支持相关tricks。

Pytorch-NLU，一个中文文本分类、序列标注工具包，支持中文长文本、短文本的多类、多标签分类任务，支持中文命名实体识别、词性标注、分词等序列标注任务。 Ptorch NLU, a Chinese text classification and sequence annotation toolkit, supports multi class and multi label classification tasks of Chinese long text and short text, and supports sequence annotation tasks such as Chinese named entity recognition, part of speech ta…

Yet another PyTorch implementation of Tacotron 2 with reduction factor and faster training speed.

Create a concurrent video analysis pipeline featuring multistream face and human pose detection, vehicle attribute detection, and the ability to encode multiple videos to local storage in a single stream.

Code for ACL'2021 paper WARP 🌀 Word-level Adversarial ReProgramming. Outperforming `GPT-3` on SuperGLUE Few-Shot text classification. https://aclanthology.org/2021.acl-long.381/

Easily train or fine-tune SOTA computer vision models with one open source training library

Observe workers as they pass in front of a camera to determine if they have adequate safety protection.

An open source implementation of CLIP.

Pretrained GANs + VAEs + classifiers for MNIST/CIFAR in pytorch.

High-resolution breast cancer screening with multi-view deep convolutional neural networks

Predict performance issues with manufacturing equipment motors. Perform local or cloud analytics of the issues found, and then display the data on a user interface to determine when failures might arise.

this is roberta wwm base distilled model which was distilled from roberta wwm by roberta wwm large

Presenting Collection of Pretrained Models. Links to pretrained models in NLP and voice.

Meta-repository of screening mammography classifiers

Improving Disfluency Detection by Self-Training a Self-Attentive Model

Monitor mechanical bolts as they move down a conveyor belt. When a bolt of an irregular size is detected, this solution emits an alert.

Protein Language Model

Build an application that alerts you when someone enters a restricted area. Learn how to use models for multiclass object detection.

73.6% GhostNet 1.0x pre-trained model on ImageNet

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes

An Out-of-the-Box Replication of GANimation using PyTorch, pretrained weights are available!

GAP-text2SQL: Learning Contextual Representations for Semantic Parsing with Generation-Augmented Pre-Training

A codebase & model zoo for pretrained backbone based on MegEngine.

中文语音识别系列，读者可以借助它快速训练属于自己的中文语音识别模型，或直接使用预训练模型测试效果。

Learning To Compare For Text , Few shot learning in text classification

Address book for computer vision models.

Artificial intelligence/machine learning course at UCF in Spring 2020 (Fall 2019 and Spring 2019)

A Robustly Optimized BERT Pretraining Approach for Vietnamese