arthurdjn commited on
Commit
70bda74
·
verified ·
1 Parent(s): 13058fc

Update README.md and model.safetensors for randlanet.toronto3d.open3d

Browse files
Files changed (2) hide show
  1. README.md +119 -0
  2. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: torch-pointcloud
4
+ tags:
5
+ - point-cloud
6
+ - 3d
7
+ - pytorch
8
+ - torch-pointcloud
9
+ - randlanet
10
+ - segmentation
11
+ datasets:
12
+ - toronto3d
13
+ model-index:
14
+ - name: randlanet.toronto3d.open3d
15
+ results:
16
+ - task:
17
+ type: point-cloud-segmentation
18
+ dataset:
19
+ name: Toronto-3D
20
+ type: toronto3d
21
+ metrics:
22
+ - name: mIoU
23
+ type: mean_iou
24
+ value: 73.88
25
+ - name: OA
26
+ type: accuracy
27
+ value: 95.85
28
+ ---
29
+
30
+ # Model card for randlanet.toronto3d.open3d
31
+
32
+ A RandLA-Net point cloud semantic segmentation model (random sampling with local feature aggregation). Trained on Toronto-3D.
33
+
34
+ ## Model Details
35
+
36
+ - **Model Type:** Point cloud semantic segmentation
37
+ - **Model Stats:**
38
+ - Params (M): 5.0
39
+ - Input channels: 6
40
+ - Classes: 8
41
+ - Features: 32
42
+ - **Dataset:** Toronto-3D
43
+ - **Metrics:** mIoU 73.88, OA 95.85
44
+ - **Paper:** [RandLA-Net: Efficient Semantic Segmentation of Large-Scale Point Clouds](https://arxiv.org/abs/1911.11236)
45
+ - **Converted from:** [isl-org/Open3D-ML](https://github.com/isl-org/Open3D-ML) (MIT)
46
+ - **Library:** [torch-pointcloud](https://github.com/arthurdjn/pytorch-pointcloud)
47
+
48
+ ## Install
49
+
50
+ ```bash
51
+ pip install torch-pointcloud
52
+ ```
53
+
54
+ ## Usage
55
+
56
+ ```python
57
+ import torch
58
+ import torch_pointcloud as tp
59
+ from torch_pointcloud.utils.data import collate
60
+
61
+ model, info = tp.create_model(
62
+ "randlanet.toronto3d.open3d",
63
+ pretrained=True,
64
+ return_info=True,
65
+ )
66
+ model = model.eval()
67
+
68
+ # synthetic sample with the keys a dataset provides
69
+ num_points = 8192
70
+ sample = {
71
+ "pos": torch.randn(num_points, 3),
72
+ "color": torch.rand(num_points, 3) * 255,
73
+ "intensity": torch.rand(num_points, 1),
74
+ "gps_time": torch.zeros(num_points, 1),
75
+ "segment": torch.zeros(num_points, dtype=torch.long),
76
+ }
77
+ data = info["transform"](sample)
78
+ data = collate([data])
79
+
80
+ with torch.no_grad():
81
+ logits = model(data.get("x"), data["pos"], data["batch"])
82
+ ```
83
+
84
+ ## Feature extraction
85
+
86
+ ```python
87
+ with torch.no_grad():
88
+ features = model.forward_features(data.get("x"), data["pos"], data["batch"])
89
+
90
+ model.reset_classifier(num_classes=0)
91
+ with torch.no_grad():
92
+ features = model(data.get("x"), data["pos"], data["batch"]) # (N, 32)
93
+ ```
94
+
95
+ ## Citation
96
+
97
+ ```bibtex
98
+ @inproceedings{hu2020randlanet,
99
+ title = {RandLA-Net: Efficient Semantic Segmentation of Large-Scale Point Clouds},
100
+ author = {Qingyong Hu and Bo Yang and Linhai Xie and Stefano Rosa and Yulan Guo and Zhihua Wang and Niki Trigoni and Andrew Markham},
101
+ booktitle = {CVPR},
102
+ year = {2020}
103
+ }
104
+
105
+ @inproceedings{tan2020toronto3d,
106
+ title = {Toronto-3D: A Large-scale Mobile LiDAR Dataset for Semantic Segmentation of Urban Roadways},
107
+ author = {Weikai Tan and Nannan Qin and Lingfei Ma and Ying Li and Jing Du and Guorong Cai and Ke Yang and Jonathan Li},
108
+ booktitle = {CVPR Workshops},
109
+ year = {2020}
110
+ }
111
+
112
+ @software{dujardin2026pytorchpointcloud,
113
+ author = {Arthur Dujardin},
114
+ title = {PyTorch PointCloud},
115
+ year = {2026},
116
+ doi = {10.5281/zenodo.22159632},
117
+ url = {https://github.com/arthurdjn/pytorch-pointcloud},
118
+ }
119
+ ```
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1859c1cb416fcfd0452fdf391b32b1342f4ffc949f1ce0e2b4fda68b537f5b8a
3
+ size 20079880