Upload learn_region_grow/__init__.py

learn_region_grow/__init__.py

@@ -0,0 +1,35 @@
+"""
+learn_region_grow
+=================
+
+A PyTorch implementation of LRGNet: Learnable Region Growing for
+Class-Agnostic Point Cloud Segmentation.
+
+Paper: LRGNet: Learnable Region Growing for Class-Agnostic Point Cloud Segmentation
+Authors: Jingdao Chen, Zsolt Kira, Yong K. Cho
+Venue: IEEE Robotics and Automation Letters (RAL), 2021
+arXiv: https://arxiv.org/abs/2103.09160
+
+Original repository: https://github.com/jingdao/learn_region_grow
+
+Modules
+-------
+- io          : Load PLY / PCD point clouds and save segmented results.
+- preprocess  : Voxel equalization, normal estimation, curvature, feature vector.
+- lrg_net     : LrgNet dual-branch 1D PointNet in PyTorch.
+- growing     : Learned region growing inference engine.
+- stage_data  : Generate supervised training examples from labeled point clouds.
+- train       : Training loop with Adam + optional cross-domain augmentation.
+- utils       : Sampling, centering, metrics, color maps.
+"""
+
+__version__ = "0.1.0"
+__author__ = "bdck"
+
+from . import io
+from . import preprocess
+from . import lrg_net
+from . import growing
+from . import stage_data
+from . import train
+from . import utils