Stable Diffusion web UI

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Models and examples built with TensorFlow

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

PyTorch Tutorial for Deep Learning Researchers

OpenMMLab Detection Toolbox and Benchmark

⚡ A Fast, Extensible Progress Bar for Python and CLI

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

Graph Neural Network Library for PyTorch

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

End-to-End Object Detection with Transformers

Python package built to ease deep learning on graph, on top of existing DL frameworks.

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

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

A collaboration friendly studio for NeRFs

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Object detection, 3D detection, and pose estimation using center point detection:

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

Neural style in TensorFlow! 🎨

Build Graph Nets in Tensorflow

Graph Convolutional Networks in PyTorch

A simple, fully convolutional model for real-time instance segmentation.