octree-nn / ocnn-pytorch

Octree-based 3D Convolutional Neural Networks
MIT License
156 stars 17 forks source link

OCNN CUDA-ops #34

Closed ChunyuFeng closed 6 months ago

ChunyuFeng commented 6 months ago

Thanks for your great job! I'm working on model deployment on jetson agx orin by tensorRT, and some octree ops (octree-conv/octree-deconv etc.) are used in my model. And I'm wondering that are those ops having a CUDA version? Thanks again!

wang-ps commented 6 months ago

The original version of ocnn was implemented with CUDA. Hope it may be helpful to you.

https://github.com/microsoft/O-CNN