Let us control diffusion models!

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

A collaboration friendly studio for NeRFs

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

Stable Diffusion built-in to Blender

🚀 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

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

[ICLR 2024] Fine-tuning LLaMA to follow Instructions within 1 Hour and 1.2M Parameters

Infinite Photorealistic Worlds using Procedural Generation

DUSt3R: Geometric 3D Vision Made Easy

The Open Images dataset

The official PyTorch implementation of Towards Fast, Accurate and Stable 3D Dense Face Alignment, ECCV 2020.

A procedural Blender pipeline for photorealistic training image generation

Simultaneous object detection and tracking using center points.

[CVPR 2024 - Oral, Best Paper Award Candidate] Marigold: Repurposing Diffusion-Based Image Generators for Monocular Depth Estimation

Pytorch implementation of RetinaNet object detection.

pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593

Automated dense category annotation engine that serves as the initial semantic labeling for the Segment Anything dataset (SA-1B).

A pytorch CUDA extension implementation of instant-ngp (sdf and nerf), with a GUI.

This is the code for Deformable Neural Radiance Fields, a.k.a. Nerfies.

An awesome PyTorch NeRF library

Implementations of few-shot object detection benchmarks

(CVPR 2021 Oral) Open World Object Detection

Pure PyTorch Implementation of NVIDIA paper on Instant Training of Neural Graphics primitives: https://nvlabs.github.io/instant-ngp/

A pytorch implementation for paper 'Exploring Simple Siamese Representation Learning'

Code release for DS-NeRF (Depth-supervised Neural Radiance Fields)

(ICCV'19) Indices Matter: Learning to Index for Deep Image Matting

Live Demo：https://modelscope.cn/studios/Damo_XR_Lab/3D_AIGC