ShoufaChen

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

segment-anything

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything

accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

GroundingDINO

[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"

VideoCrafter

VideoCrafter2: Overcoming Data Limitations for High-Quality Video Diffusion Models

InternGPT

InternGPT (iGPT) is an open source demo platform where you can easily showcase your AI models. Now it supports DragGAN, ChatGPT, ImageBind, multimodal chat like GPT-4, SAM, interactive image editing, etc. Try it at igpt.opengvlab.com (支持DragGAN、ChatGPT、ImageBind、SAM的在线Demo系统)

Awesome-Video-Diffusion

A curated list of recent diffusion models for video generation, editing, restoration, understanding, etc.

InternImage

[CVPR 2023 Highlight] InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions

EVA

EVA Series: Visual Representation Fantasies from BAAI

DiffusionDet

[ICCV2023 Best Paper Finalist] PyTorch implementation of DiffusionDet (https://arxiv.org/abs/2211.09788)

Awesome-Anything

General AI methods for Anything: AnyObject, AnyGeneration, AnyModel, AnyTask, AnyX

torchdrug

A powerful and flexible machine learning platform for drug discovery

OpenDiT

OpenDiT: An Easy, Fast and Memory-Efficient System for DiT Training and Inference

DiffDock

Implementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking

SparseInst

[CVPR 2022] SparseInst: Sparse Instance Activation for Real-Time Instance Segmentation

Groma

[ECCV2024] Grounded Multimodal Large Language Model with Localized Visual Tokenization

DAIR-V2X

VLPart

[ICCV2023] VLPart: Going Denser with Open-Vocabulary Part Segmentation

FlashFace

CycleMLP

[ICLR'22 Oral] Implementation of "CycleMLP: A MLP-like Architecture for Dense Prediction"

DDQ

Dense Distinct Query for End-to-End Object Detection (CVPR2023)

ByteTrack-V2

Higra

Hierarchical Graph Analysis

MetaBEV

MetaBEV: Solving Sensor Failures for BEV Detection and Map Segmentation

Awesome-Pretrained-RL

Intra-and-Inter-Human-Relation-Network-for-MPEE

I²R-Net: Intra- and Inter-Human Relation Network for Multi-Person Pose Estimation

CtrlFormer_robotic

Implantation of CtrlFormer

SpatialDM

Spatial direct messaging detected by bivariate Moran

litegt

[CIKM-21] Pytorch implementation of LiteGT: Efficient and Lightweight Graph Transformers