MIC-DKFZ/image_classification

A visual-based graph node editor for training computer vision models.

Datasets, Transforms and Models specific to Computer Vision

All-in-one training for vision models (YOLO, ViTs, RT-DETR, DINOv3): pretraining, fine-tuning, distillation.

A python library for self-supervised learning on images.

OpenMMLab Pre-training Toolbox and Benchmark

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

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

CVNets: A library for training computer vision networks

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM, Finer-CAM, LeGrad, RefineCAM)