Repository Index

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

• ultralytics
• yolov8
• yolo
• deep-learning

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

• yolov5
• object-detection
• pytorch
• onnx

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

• pytorch
• resnet
• pretrained-models
• pretrained-weights

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

• deep-learning
• pytorch
• grad-cam
• visualizations

An absolute beginner's guide to Machine Learning and Image Classification with Neural Networks

• neural-network
• image-classification
• machine-learning
• caffe

Accelerated deep learning R&D

• deep-learning
• reinforcement-learning
• machine-learning
• computer-vision

Official PyTorch implementation of VisionHOPE: Visual Backbones as Self-Modifying Learning Systems.

• backbone
• computer-vision
• image-classification
• backbone-networks

AgML is a centralized framework for agricultural machine learning. AgML provides access to public agricultural datasets for common agricultural deep learning tasks, with standard benchmarks and pretrained models, as well the ability to generate synthetic data and annotations.

• deep-learning
• agriculture
• pytorch
• dataset

A simple, flexible, and extensible template for PyTorch. It's beautiful.

• deep-learning
• computer-vision
• cnn
• pytorch