|
--- |
|
datasets: |
|
- RoyYang0714/3D-MOOD |
|
license: apache-2.0 |
|
pipeline_tag: image-to-3d |
|
--- |
|
|
|
<div align="center"> |
|
|
|
# 3D-MOOD: Lifting 2D to 3D for Monocular Open-Set Object Detection |
|
|
|
<a href="https://huggingface.co/papers/2507.23567"><img src='https://img.shields.io/badge/arXiv-Paper-red?logo=arxiv&logoColor=white' alt='arXiv'></a> |
|
<a href='https://royyang0714.github.io/3D-MOOD'><img src='https://img.shields.io/badge/Project%20Page-Website-green?logo=googlechrome&logoColor=white' alt='Project Page'></a> |
|
<a href='https://huggingface.co/spaces/RoyYang0714/3D-MOOD'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Live%20Demo-blue'></a> \ |
|
<a href='https://huggingface.co/RoyYang0714/3D-MOOD'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Models-blue'></a> |
|
<a href='https://huggingface.co/datasets/RoyYang0714/3D-MOOD'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Datasets-blue'></a> |
|
|
|
</div> |
|
|
|
<div> |
|
<img src="https://github.com/cvg/3D-MOOD/raw/main/assets/overview.png" width="100%" alt="Banner 2" align="center"> |
|
</div> |
|
|
|
This repository contains the official models and code for the paper [3D-MOOD: Lifting 2D to 3D for Monocular Open-Set Object Detection](https://huggingface.co/papers/2507.23567). |
|
|
|
Project Page: https://royyang0714.github.io/3D-MOOD |
|
Code: https://github.com/cvg/3D-MOOD |
|
|
|
## Abstract |
|
|
|
Monocular 3D object detection is valuable for various applications such as robotics and AR/VR. Existing methods are confined to closed-set settings, where the training and testing sets consist of the same scenes and/or object categories. However, real-world applications often introduce new environments and novel object categories, posing a challenge to these methods. In this paper, we address monocular 3D object detection in an open-set setting and introduce the first end-to-end 3D Monocular Open-set Object Detector (3D-MOOD). We propose to lift the open-set 2D detection into 3D space through our designed 3D bounding box head, enabling end-to-end joint training for both 2D and 3D tasks to yield better overall performance. We condition the object queries with geometry prior and overcome the generalization for 3D estimation across diverse scenes. To further improve performance, we design the canonical image space for more efficient cross-dataset training. We evaluate 3D-MOOD on both closed-set settings (Omni3D) and open-set settings (Omni3D to Argoverse 2, ScanNet), and achieve new state-of-the-art results. |
|
|
|
## Demo |
|
|
|
We provide the [`demo.py`](https://github.com/cvg/3D-MOOD/blob/main/scripts/demo.py) to test whether the installation is complete. |
|
|
|
```bash |
|
python scripts/demo.py |
|
``` |
|
|
|
It will save the prediction as follow to `assets/demo/output.png`. |
|
|
|
<div align="center"> |
|
<img src="https://github.com/cvg/3D-MOOD/raw/main/assets/demo/output.png" alt="Demo Output"> |
|
</div> |
|
|
|
You can also try the live demo on [Hugging Face Spaces](https://huggingface.co/spaces/RoyYang0714/3D-MOOD)! |
|
|
|
## Citation |
|
|
|
If you find our work useful in your research please consider citing our publications: |
|
```bibtex |
|
@article{yang20253d, |
|
title={3D-MOOD: Lifting 2D to 3D for Monocular Open-Set Object Detection}, |
|
author={Yang, Yung-Hsu and Piccinelli, Luigi and Segu, Mattia and Li, Siyuan and Huang, Rui and Fu, Yuqian and Pollefeys, Marc and Blum, Hermann and Bauer, Zuria}, |
|
journal={arXiv preprint arXiv:2507.23567}, |
|
year={2025} |
|
} |
|
``` |