CCInc / 3d-ml

A versatile framework for 3D machine learning built on Pytorch Lightning and Hydra [looking for contributors!]
15 stars 3 forks source link

Add Dataset Transforms Pipeline #7

Closed CCInc closed 2 years ago

CCInc commented 2 years ago

Add support for Pytorch Geometric data transformation pipelines for train, test, and validation datasets. Reimplement some TP3D transform methods within the new repo, such as AddFeatsByKeys and GridSampling3d. Investigate using OpenPoints transforms.