ATF-3D: Semi-Supervised 3D Object Detection With Adaptive Thresholds Filtering Based on Confidence and Distance

Zehan Zhang,Yang Ji,Wei Cui,Yulong Wang,Hao Li,Xian Zhao,Duo Li,Sanli Tang,Ming Yang,Wenming Tan,Shiliang Pu
DOI: https://doi.org/10.1109/lra.2022.3187496
IF: 5.2
2022-10-01
IEEE Robotics and Automation Letters
Abstract:Performance of current point cloud-based outdoor 3D object detection relies heavily on large-scale high-quality 3D annotations. However, such annotations are usually expensive to collect and outdoor scenes easily accumulate massive unlabeled data containing rich scenes. Semi-supervised learning is a effective alternative to utilize both labeled and unlabeled data, but remains unexplored in outdoor 3D object detection. Inspired by indoor semi-supervised 3D detection methods, SESS and 3DIoUMatch, we propose ATF-3D, a semi-supervised 3D object detection framework for outdoor scenes. Specifically, we design a simple yet effective adaptive thresholds search method based on distances and categories for obtaining high-quality pseudo labels. Concurrently, we propose an iterative training mechanism with pseudo-label training and self-ensembling learning to combine the advantages of both schemes. Furthermore, we adopt point cloud data augmentations in the self-ensembling learning stage to further improve the performance. Our ATF-3D ranks first among all single-model methods in the ONCE benchmark. Results on both ONCE and Waymo datasets demonstrate substatial improvements over the supervised baseline.
robotics
What problem does this paper attempt to address?