Imagenet Python Packages

timm

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

segmentation-models-pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

efficientnet-pytorch

A PyTorch implementation of EfficientNet

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

pretrainedmodels

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

cvat-sdk

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling services, for image, video, and 3D annotation with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

datumaro

Dataset Management Framework, a Python library and a CLI tool to build, analyze and manage Computer Vision datasets.

labelimg

LabelImg is now part of the Label Studio community. The popular image annotation tool created by Tzutalin is no longer actively being developed, but you can check out Label Studio, the open source data labeling tool for images, text, hypertext, audio, video and time-series data.

image-classifiers

Classification models trained on ImageNet. Keras.

cvat-cli

Computer Vision Annotation Tool (CVAT) is a leading platform for building high-quality visual datasets for vision AI. It offers open-source, cloud, and enterprise products, as well as labeling services, for image, video, and 3D annotation with AI-assisted labeling, quality assurance, team collaboration, analytics, and developer APIs.

keras-cv-attention-models

Keras beit,caformer,CMT,CoAtNet,convnext,davit,dino,efficientdet,edgenext,efficientformer,efficientnet,eva,fasternet,fastervit,fastvit,flexivit,gcvit,ghostnet,gpvit,hornet,hiera,iformer,inceptionnext,lcnet,levit,maxvit,mobilevit,moganet,nat,nfnets,pvt,swin,tinynet,tinyvit,uniformer,volo,vanillanet,yolor,yolov7,yolov8,yolox,gpt2,llama2, alias kecam

dataflow-cv

🌊 Everything your model doesn't do. Convert, visualize, evaluate, and more — a single CLI for all CV data.

tensorlayer

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

keras-unet-collection

The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.

torchdistill

A coding-free framework built on PyTorch for reproducible deep learning studies. PyTorch Ecosystem. 🏆26 knowledge distillation methods presented at TPAMI, CVPR, ICLR, ECCV, NeurIPS, ICCV, AAAI, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.

chainercv2

Sandbox for training deep learning networks

gluoncv2

Sandbox for training deep learning networks

kerascv

Sandbox for training deep learning networks

gcvit

Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer

imagenet-c

Corruption and Perturbation Robustness (ICLR 2019)

tvmodels

Pytorch implementation of vision models.

efficientvit-gml

Efficient vision foundation models for high-resolution generation and perception.

imagenetscraper

👁 Bulk-download all thumbnails from an ImageNet synset, with optional rescaling

tensorflowcv

Sandbox for training deep learning networks