Stars
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
OpenMMLab Detection Toolbox and Benchmark
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
End-to-End Object Detection with Transformers
🐍 Geometric Computer Vision Library for Spatial AI
TensorFlow CNN for fast style transfer ⚡🖥🎨🖼
Refine high-quality datasets and visual AI models
Fast and Accurate ML in 3 Lines of Code
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
Text-to-3D & Image-to-3D & Mesh Exportation with NeRF + Diffusion.
Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also …
OpenMMLab's next-generation platform for general 3D object detection.
OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inference
[ECCV 2022] This is the official implementation of BEVFormer, a camera-only framework for autonomous driving perception, e.g., 3D object detection and semantic map segmentation.
Deformable DETR: Deformable Transformers for End-to-End Object Detection.
A library for scientific machine learning and physics-informed learning