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

pretrainedmodels

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

cvat-sdk

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

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.

datumaro

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

image-classifiers

Classification models trained on ImageNet. Keras.

cvat-cli

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

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

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.

tensorlayer

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

d3m-segmentation-models-pytorch

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

segmentation-models-pytorch-deepflash2

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

gluoncv2

Sandbox for training deep learning networks

chainercv2

Sandbox for training deep learning networks

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.

kerascv

Image classification models for Keras

mindcv

A toolbox of vision models and algorithms based on MindSpore

tensorflowcv

Image classification models for TensorFlow

gcvit

Tensorflow 2.0 Implementation of GCViT: Global Context Vision Transformer

mlclf

mini-imagenet and tiny-imagent dataset transformation for traditional classification task and also for the format for few-shot learning / meta-learning tasks

dataflow-cv

DataFlow-CV: Where Vibe Coding meets CV data. 🌊 Convert & visualize data.