An open-source project dedicated to tracking and segmenting any objects in videos, either automatically or interactively. The primary algorithms utilized include the Segment Anything Model (SAM) fo…

pix2pix3D: Generating 3D Objects from 2D User Inputs

Modeling SARS-CoV-2 antigen escape from antibodies using pretrained protein language model embeddings.

The software used in NeurIPS 2022 Paper Don't Pour Cereal into Coffee: Differentiable Temporal Logic for Temporal Action Segmentation.

This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.

GREATER and CARLA - Revealing Occlusions with 4D Neural Fields (CVPR 2022) - Official Implementation

[ECCV2022] Masked Autoencoders for Point Cloud Self-supervised Learning

code for our work: RFNet for Incomplete Multi-modal Brain Tumor Segmentation

Code, dataset and models for our CVPR 2022 publication "Text2Pos"

[ICCV 2021] A dataset of non-rigidly deforming objects.

PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.

A Diagnostic Dataset for Compositional Language and Elementary Visual Reasoning

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Self-Supervised Learning for Domain Adaptation on Point-Clouds

Unsupervised Abstract Reasoning for Raven’s Problem Matrices

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

[ECCV 2020] ScanRefer: 3D Object Localization in RGB-D Scans using Natural Language

[CVPR 2021] Exploiting Edge-Oriented Reasoning for 3D Point-based Scene Graph Analysis (official pytorch implementation)

Span-based Localizing Network for Natural Language Video Localization (ACL 2020)

Mayavi tutorial material

Chamfer Distance in Pytorch with f-score

A list of papers and datasets about point cloud analysis (processing) since 2017. Update every day!

Pytorch package to compute Chamfer distance between point sets (pointclouds).

PyTorch Wrapper for Earth-Mover-Distance (EMD) for 3D point cloud regression

Collection of reinforcement learning algorithms

Planning, inverse planning, and inference in planning, using PDDL and Gen.

robosuite: A Modular Simulation Framework and Benchmark for Robot Learning

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