← all repositories

lucidrains/point-transformer-pytorch

A PyTorch implementation of the Point Transformer self-attention layer for 3D point cloud processing.

601 stars Python Computer VisionML Frameworks
point-transformer-pytorch
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the Point Transformer layer, based on a 2020 arXiv paper. The layer implements vector attention specifically designed for point cloud data, enabling tasks like classification and segmentation. It supports configurable hidden dimensions for position and attention MLPs, and optional k-nearest neighbor constraints to reduce computational cost by limiting attention to nearby points.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.