Stable Diffusion web UI

🤗 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.

🙌 OpenHands: Code Less, Make More

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Universal LLM Deployment Engine with ML Compilation

Fast and memory-efficient exact attention

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

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

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 Collection of Variational Autoencoders (VAE) in PyTorch.

[CVPR 2024] Real-Time Open-Vocabulary Object Detection

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

Official PyTorch implementation of SegFormer

The official Python client for the Hugging Face Hub.

Official Repo For "Sa2VA: Marrying SAM2 with LLaVA for Dense Grounded Understanding of Images and Videos"

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on d…

An elegent pytorch implement of transformers

[NeurIPS 2022] Denoising Diffusion Restoration Models -- Official Code Repository

Inflated i3d network with inception backbone, weights transfered from tensorflow

Official PyTorch implementation of "Multi-modal Queried Object Detection in the Wild" (accepted by NeurIPS 2023)

3D image classification using CNN (Convolutional Neural Network)

[ICCV2023] DiffuMask: Synthesizing Images with Pixel-level Annotations for Semantic Segmentation Using Diffusion Models

Official Implementation of LARGE: Latent-Based Regression through GAN Semantics

A GAN inversion toolbox based on PyTorch library. We design a unified pipeline for inversion methods and conduct a comprehensive benchmark.

12306tickets

基于 Tornado 6.x 的 RESTfulAPI 风格的项目模板，用于快速构建企业级高性能、高并发的服务端。

StyleGAN2 - Official TensorFlow Implementation