357 Open source projects that are alternatives of or similar to Pytorch Image Models

Detecting cinema shot types using a ResNet-50

Machine learning, in numpy

Training a ResNet on UMDFaces for face recognition

train resnet on imagenet from scratch with caffe

🙄 Difficult algorithm, Simple code.

PyTorch implementation of DQN, AC, ACER, A2C, A3C, PG, DDPG, TRPO, PPO, SAC, TD3 and ....

PyTorch implementations of recent Computer Vision tricks

PyTorch implementation of HyperNetworks (Ha et al., ICLR 2017) for ResNet (Residual Networks)

Residual networks implementation using Keras-1.0 functional API

Classification with PyTorch.

A one stop shop for all of your activity recognition needs.

Caffe models (including classification, detection and segmentation) and deploy files for famouse networks

Reference implementation for Blueprint Separable Convolutions (CVPR 2020)

Implementing YOLO using ResNet as the feature extraction network

State-of-the-art detector of Boring hats in images and videos.

Tools to Design or Visualize Architecture of Neural Network

Gluon to PyTorch deep neural network model converter

Learning Fine-grained Image Similarity with Deep Ranking is a novel application of neural networks, where the authors use a new multi scale architecture combined with a triplet loss to create a neural network that is able to perform image search. This repository is a simplified implementation of the same

a lightweight image matting model

Official PyTorch implementation of "Artist Style Transfer Via Quadratic Potential"

Code for 15th place in Kaggle Google AI Open Images - Object Detection Track

[CVPR 2020 Oral] PyTorch implementation of "Disentangling and Unifying Graph Convolutions for Skeleton-Based Action Recognition"

Pretrained TorchVision models on CIFAR10 dataset (with weights)

EMNLP 2020: "Dialogue Response Ranking Training with Large-Scale Human Feedback Data"

Awesome Pretrained Chinese NLP Models，高质量中文预训练模型集合

Modification of convolutional neural net "UNET" for image segmentation in Keras framework

This project implements a real-time image and video object detection classifier using pretrained yolov3 models.

CRSLab is an open-source toolkit for building Conversational Recommender System (CRS).

Optimus: the first large-scale pre-trained VAE language model

Sandbox for training deep learning networks

Named Entity Recognition with BERT using TensorFlow 2.0

Generic Foreground Segmentation in Images

A curated list of pretrained sentence and word embedding models

Porting of Skip-Thoughts pretrained models from Theano to PyTorch & Torch7

Pre-Trained Models for ToD-BERT

中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard

Semantic text segmentation. For sentence boundary detection, compound splitting and more.

A fast, high-quality neural vocoder.

DiffWave is a fast, high-quality neural vocoder and waveform synthesizer.

High-quality Neural Networks for Computer Vision 😎

Breast density classification with deep convolutional neural networks

中文 预训练 ELECTRA 模型: 基于对抗学习 pretrain Chinese Model

Pretrained is the most complete and frequently updated list of pretrained top-performing models. Tensorflow, Theano and others. Want to add your model? File an issue, and we will add it.

A TensorFlow implementation of the Mnemonic Descent Method.

Motion Fused Frames implementation in PyTorch, codes and pretrained models.

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

DimeNet and DimeNet++ models, as proposed in "Directional Message Passing for Molecular Graphs" (ICLR 2020) and "Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Molecules" (NeurIPS-W 2020)

Pretrained BERT model for analysing COVID-19 Twitter data

[ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions

Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

This repository contains PyTorch implementation for the baseline models from the paper Utterance-level Dialogue Understanding: An Empirical Study

🏡 Fast & easy transfer learning for NLP. Harvesting language models for the industry. Focus on Question Answering.

Create a smart video application using the Intel Distribution of OpenVINO toolkit. The toolkit uses models and inference to run single-class object detection.

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

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）

⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥