CAT++: Enhancing 3D Annotations with Hierarchical-Interleaved Encoding and Attention-Conditioned Implicit Representation
摘要
This work presents CAT++, an enhanced Context-Aware Transformer (CAT), to address the need for more effective 3D point cloud annotation. CAT++ is an end-to-end 3D automatic annotator that lifts 2D weak annotations into 3D, thereby minimizing human annotation burdens. Specifically, it employs a hierarchical-interleaved encoder (HIE) to enhance its encoder-decoder architecture. The HIE alternates between intra- and inter-object Transformer blocks, integrating local and global object features by performing self-attentions along the sequence and batch dimensions. This process continuously combines local and global object features and creates a 3D representation that links interactions between objects at different scales, leading to a more comprehensive understanding of the scene. Moreover, CAT++ features an attention-conditioned implicit neural representation (INR) network, which has been specially engineered to enable more precise and efficient continuous modeling of 3D object surfaces. This improves CAT++’s multi-task learning and its ability to handle hard samples. Results on KITTI and nuScenes datasets show CAT++’s state-of-the-art performance, outperforming other methods by up to 2%