arXiv reaDer
SSPC-Net:半教師ありセマンティック3Dポイントクラウドセグメンテーションネットワーク
SSPC-Net: Semi-supervised Semantic 3D Point Cloud Segmentation Network
点群のセマンティックセグメンテーションは、3Dシーンを理解する上で重要なタスクです。既存の方法は、主に、教師ありセマンティックセグメンテーションに多数の注釈付きラベルを使用することに重点を置いています。それにもかかわらず、監視されたセグメンテーションタスクのためにそのような大きな点群に手動でラベルを付けることは時間がかかります。注釈付きラベルの数を減らすために、SSPC-Netという名前の半教師ありセマンティックポイントクラウドセグメンテーションネットワークを提案します。ここでは、いくつかの注釈付き3Dポイントからラベルなしポイントのラベルを推測することによってセマンティックセグメンテーションネットワークをトレーニングします。私たちの方法では、最初に点群全体をスーパーポイントに分割し、スーパーポイントグラフを作成して、点群の長距離依存関係をマイニングします。次に、構築されたスーパーポイントグラフに基づいて、教師なしスーパーポイントの疑似ラベルを生成する動的ラベル伝播方法を開発します。特に、生成された疑似ラベルを動的に選択するために、スーパーポイントドロップアウト戦略を採用しています。教師なしスーパーポイントの生成された疑似ラベルを完全に活用するために、スーパーポイント特徴埋め込みのための結合注意メカニズムをさらに提案します。最後に、クロスエントロピー損失を使用して、教師なしスーパーポイントのラベルと教師なしスーパーポイントの疑似ラベルを使用してセマンティックセグメンテーションネットワークをトレーニングします。さまざまなデータセットでの実験により、半教師ありセグメンテーション方法は、注釈付き3Dポイントが少ない現在の半教師ありセグメンテーション方法よりも優れたパフォーマンスを実現できることが実証されています。私たちのコードはhttps://github.com/MMCheng/SSPC-Netで入手できます。
Point cloud semantic segmentation is a crucial task in 3D scene understanding. Existing methods mainly focus on employing a large number of annotated labels for supervised semantic segmentation. Nonetheless, manually labeling such large point clouds for the supervised segmentation task is time-consuming. In order to reduce the number of annotated labels, we propose a semi-supervised semantic point cloud segmentation network, named SSPC-Net, where we train the semantic segmentation network by inferring the labels of unlabeled points from the few annotated 3D points. In our method, we first partition the whole point cloud into superpoints and build superpoint graphs to mine the long-range dependencies in point clouds. Based on the constructed superpoint graph, we then develop a dynamic label propagation method to generate the pseudo labels for the unsupervised superpoints. Particularly, we adopt a superpoint dropout strategy to dynamically select the generated pseudo labels. In order to fully exploit the generated pseudo labels of the unsupervised superpoints, we furthermore propose a coupled attention mechanism for superpoint feature embedding. Finally, we employ the cross-entropy loss to train the semantic segmentation network with the labels of the supervised superpoints and the pseudo labels of the unsupervised superpoints. Experiments on various datasets demonstrate that our semi-supervised segmentation method can achieve better performance than the current semi-supervised segmentation method with fewer annotated 3D points. Our code is available at https://github.com/MMCheng/SSPC-Net.
updated: Mon May 24 2021 14:04:18 GMT+0000 (UTC)
published: Fri Apr 16 2021 02:37:27 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト