データ拡張は、画像認識タスクのパフォーマンスの向上に大きく貢献しており、関連する多くの研究が行われています。ただし、3Dポイントクラウドデータのデータ拡張についてはあまり検討されていません。 3Dラベルは、2Dラベルよりも洗練された豊富な構造情報を備えているため、より多様で効果的なデータ拡張が可能になります。本論文では、3Dラベルの豊富な情報をより有効に活用して3Dオブジェクト検出器の性能を向上させることができる部分認識データ拡張(PA-AUG)を提案します。 PA-AUGは、オブジェクトをパーティションに分割し、確率的に5つの拡張メソッドを各ローカル領域に適用します。これは、既存の点群データ拡張方法と互換性があり、検出器のアーキテクチャに関係なく、普遍的に使用できます。 PA-AUGは、KITTIデータセットのすべてのクラスの最先端の3Dオブジェクト検出器のパフォーマンスを改善し、列車データを約2.5倍に増やすという同等の効果があります。また、PA-AUGは、特定のデータセットのパフォーマンスを向上させるだけでなく、破損したデータに対しても堅牢であることも示しています。コードはhttps://github.com/sky77764/pa-aug.pytorchで入手できます。
Data augmentation has greatly contributed to improving the performance in image recognition tasks, and a lot of related studies have been conducted. However, data augmentation on 3D point cloud data has not been much explored. 3D label has more sophisticated and rich structural information than the 2D label, so it enables more diverse and effective data augmentation. In this paper, we propose part-aware data augmentation (PA-AUG) that can better utilize rich information of 3D label to enhance the performance of 3D object detectors. PA-AUG divides objects into partitions and stochastically applies five augmentation methods to each local region. It is compatible with existing point cloud data augmentation methods and can be used universally regardless of the detector's architecture. PA-AUG has improved the performance of state-of-the-art 3D object detector for all classes of the KITTI dataset and has the equivalent effect of increasing the train data by about 2.5×. We also show that PA-AUG not only increases performance for a given dataset but also is robust to corrupted data. The code is available at https://github.com/sky77764/pa-aug.pytorch