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

A list of popular deep learning models related to classification, segmentation and detection problems

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

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

Implementation of popular deep learning networks with TensorRT network definition API

Hands on Imagenet training. Unofficial ModelZoo project on Pytorch. MobileNetV3 Top1 75.64🌟 GhostNet1.3x 75.78🌟

A Multi-Class Brain Tumor Classifier using Convolutional Neural Network with 99% Accuracy achieved by applying the method of Transfer Learning using Python and Pytorch Deep Learning Framework

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

Implementation of EfficientNet model. Keras and TensorFlow Keras.

Paddle Large Scale Classification Tools，supports ArcFace, CosFace, PartialFC, Data Parallel + Model Parallel. Model includes ResNet, ViT, DeiT, FaceViT.

Gluon to PyTorch deep neural network model converter

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

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

Address book for computer vision models.

73.6% GhostNet 1.0x pre-trained model on ImageNet

Classification models trained on ImageNet. Keras.

A collection of deep learning models (PyTorch implemtation)

Pytorch Imagenet Models Example + Transfer Learning (and fine-tuning)

Concise, Modular, Human-friendly PyTorch implementation of MixNet with Pre-trained Weights.

More readable and flexible yolov5 with more backbone(resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer) and (cbam，dcn and so on), and tensorrt

Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

Fengshenbang-LM(封神榜大模型)是IDEA研究院认知计算与自然语言研究中心主导的大模型开源体系，成为中文AIGC和认知智能的基础设施。

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

Segmentation models with pretrained backbones. PyTorch.

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

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

Visual Question Answering in Pytorch

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening

Repository containing an independent implementation of the paper: "A Fully Progressive Approach to Single-Image Super-Resolution"

Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding

PyTorch implementation of CNNs for CIFAR benchmark

Pytorch-Named-Entity-Recognition-with-BERT

PyTorch and Tensorflow functional model definitions

猫狗大战

Keras implementation of BERT with pre-trained weights

A deep neural network that learns to drive in video games

A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.

The PyTorch-based audio source separation toolkit for researchers

Tensorflow implementation of the SRGAN algorithm for single image super-resolution

Tutorial for video classification/ action recognition using 3D CNN/ CNN+RNN on UCF101

Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.

This repository allows you to get started with a gui based training a State-of-the-art Deep Learning model with little to no configuration needed! NoCode training with TensorFlow has never been so easy.

OpenMMLab Image Classification Toolbox and Benchmark

Real-time pose estimation accelerated with NVIDIA TensorRT

Tensorflow ResNet implementation on cifar10

Books, Presentations, Workshops, Notebook Labs, and Model Zoo for Software Engineers and Data Scientists wanting to learn the TF.Keras Machine Learning framework

A collection of deep learning tutorials using Tensorflow and Python

Silero Models: pre-trained STT models and benchmarks made embarrassingly simple

a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results

高质量中文预训练模型集合：最先进大模型、最快小模型、相似度专门模型

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.

Training & Quantization of embedded friendly Deep Learning / Machine Learning / Computer Vision models

Train ResNet on ImageNet in Tensorflow 2.0; ResNet 在ImageNet上完整训练代码

Automated Deep Learning without ANY human intervention. 1'st Solution for AutoDL [email protected]

PyTorch to Keras model convertor

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

♾A curated list of papers and code about very deep neural networks

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

Reproduces ResNet-V3 with pytorch

R2CNN: Rotational Region CNN Based on FPN (Tensorflow)