pytorch3d icon indicating copy to clipboard operation
pytorch3d copied to clipboard

Published 20 hours ago • verified publisher icon facebookresearch

Octree Data Structure Support

Open aradhyamathur opened this issue 5 years ago • 4 comments

🚀 Feature

Support for Octrees based neural networks

Motivation

Recently several research works by Wang et al. have proposed Octree-Based CNNs for various shape analysis tasks. A similar feature implementation that allows the operations over octrees could be of great benefit especially due to the reduced memory footprint that octrees allow for 3D data.

Pitch

The current implementation exists in tensorflow only given here. It would great if pytorch3d too has similar features to allow the creation of octree based neural networks. Would be glad to contribute.

aradhyamathur avatar Oct 08 '20 17:10 aradhyamathur

This sounds like a great suggestion! We will discuss this with our team and will comment here for a timeline.

gkioxari avatar Oct 09 '20 16:10 gkioxari

It seems that two days after this conversation they added support for pytorch https://github.com/microsoft/O-CNN/tree/master/pytorch pretty efficient

albertotono avatar Nov 02 '20 22:11 albertotono

Yeah but I guess it will be better to have this data structure in pytorch3D as well, to allow some more flexibility to users to use octrees.

aradhyamathur avatar Nov 03 '20 09:11 aradhyamathur

They also confirmed that it is still under development.

albertotono avatar Nov 03 '20 14:11 albertotono