Skip to content

Home

Jump to bottom
Zhiang Chen edited this page Dec 17, 2021 · 3 revisions

Welcome to the 3d_rock_detection wiki!

Where to start

  1. PartSegmentation tutorial: https://github.com/nicolas-chaulet/torch-points3d/blob/master/notebooks/PartSegmentationKPConv.ipynb
  2. Creating your own datasets: https://torch-points3d.readthedocs.io/en/latest/src/tutorials.html
    1.0. Creating Your Own Datasets from Pytorch Geometric: https://pytorch-geometric.readthedocs.io/en/latest/notes/create_dataset.html
    1.1. How to create a shapenet dataset for torch_points3d: https://github.com/nicolas-chaulet/torch-points3d/blob/8e4c19ecb81926626231bf185e9eca77d92a0606/torch_points3d/datasets/segmentation/shapenet.py#L240

Todo

  1. create a folder "datasets/segmentation". Then under this folder, create your own dataset.

Note

self.processed_path is initiated when initiating super object InMemoryDataset. More accurate, it's the super object of InMemoryDataset, from torch_geometric.data.dataset import Dataset

Clone this wiki locally