Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

CN-RMA: Combined Network with Ray Marching Aggregation for 3D Indoors Object Detection from Multi-view Images

About

This paper introduces CN-RMA, a novel approach for 3D indoor object detection from multi-view images. We observe the key challenge as the ambiguity of image and 3D correspondence without explicit geometry to provide occlusion information. To address this issue, CN-RMA leverages the synergy of 3D reconstruction networks and 3D object detection networks, where the reconstruction network provides a rough Truncated Signed Distance Function (TSDF) and guides image features to vote to 3D space correctly in an end-to-end manner. Specifically, we associate weights to sampled points of each ray through ray marching, representing the contribution of a pixel in an image to corresponding 3D locations. Such weights are determined by the predicted signed distances so that image features vote only to regions near the reconstructed surface. Our method achieves state-of-the-art performance in 3D object detection from multi-view images, as measured by mAP@0.25 and mAP@0.5 on the ScanNet and ARKitScenes datasets. The code and models are released at https://github.com/SerCharles/CN-RMA.

Guanlin Shen, Jingwei Huang, Zhihua Hu, Bin Wang• 2024

Related benchmarks

TaskDatasetResultRank
3D Object DetectionScanNet V2 (val)
mAP@0.2558.6
361
3D Object DetectionScanNet (val)
mAP@0.2554.6
75
3D Object DetectionARKitScenes
mAP@0.2567.6
23
3D Object DetectionScanNet v2 (test)
mAP@0.2551
23
3D Object DetectionARKitScenes (test)
mAP@0.2567.6
9
3D Object DetectionScanNet (train)
Training Time (hrs)121
3
3D Object DetectionScanNet (test)
Inference Time (min)10
3
Showing 7 of 7 rows

Other info

Code

Follow for update