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

3D U-Net model for volumetric semantic segmentation written in pytorch

Official PyTorch implementation of HCCNet: Efficient Semantic Matching with Hypercolumn Correlation (WACV '24 Oral, Best paper finalist (top 0.6%))

The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)

An extension of the PyTorch library containing various tools for performing deep learning in hyperbolic space.

Repository for "Deformable 3D Convolution for Video Super-Resolution", SPL, 2020

[MICCAI 2023] DermoSegDiff: A Boundary-aware Segmentation Diffusion Model for Skin Lesion Delineation

Learning Geometry-Disentangled Representation for Complementary Understanding of 3D Object Point Cloud. (AAAI2021)

Adaptive Graph Convolution for Point Cloud Analysis

[CVPR 2024 Highlight] DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Models

[TPAMI 2022 & CVPR 2020 Oral] Dynamic Graph Message Passing Networks

Deep Learning-based Liver Vessel Separation with plug-and-play modules: Skeleton Tracking and Graph Attention

[ECCV-2020-oral]-Semantic Flow for Fast and Accurate Scene Parsing

[TNNLS 2023] The official repo for the paper "Spectral-Spatial Global Graph Reasoning for Hyperspectral Image Classification".

This is the code for Pixel-wise Sparse Graph Reasoning.

Pytorch code for view-GCN (CVPR2020)

[ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding

[TMI 2024] Geometry-Aware Attenuation Learning for Sparse-View CBCT Reconstruction

DOLCE, ICCV2023. Pytorch Implementation.

[ICCV 2023] Single-Stage Diffusion NeRF

Geometric Latent Diffusion Models for 3D Molecule Generation

Python wrapper for CurveLab's 2D and 3D curvelet transforms