Starred repositories
Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥
YOLOv6: a single-stage object detection framework dedicated to industrial applications.
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)
A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop which is flexible enough to handle the majority of use cases…
A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.
Transfer Learning Library for Domain Adaptation and Domain Generalization of Object Detection.
Easy & Modular Computer Vision Detectors, Trackers & SAM - Run YOLOv9,v8,v7,v6,v5,R,X in under 10 lines of code.
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
Convert Machine Learning Code Between Frameworks
We write your reusable computer vision tools. 💜
OpenMMLab YOLO series toolbox and benchmark. Implemented RTMDet, RTMDet-Rotated,YOLOv5, YOLOv6, YOLOv7, YOLOv8,YOLOX, PPYOLOE, etc.
Image inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.
[ICCV 2023] A latent space for stochastic diffusion models
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
A collection of resources and papers on Diffusion Models
YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quanti…
OpenMMLab Detection Toolbox and Benchmark
Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.
An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
[ICML 2023] Official PyTorch implementation of Global Context Vision Transformers
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Recent Transformer-based CV and related works.
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch