Omnidata: A Scalable Pipeline for Making Multi-Task Mid-Level Vision Datasets from 3D Scans
Not on the map. Why?
Cited by 61 in Diffusion models, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
This paper introduces a pipeline to parametrically sample and render multi-task vision datasets from comprehensive 3D scans from the real world. Changing the sampling parameters allows one to "steer" the generated datasets to emphasize specific information. In addition to enabling interesting lines of research, we show the tooling and generated data suffice to train robust vision models. Common architectures trained on a generated starter dataset reached state-of-the-art performance on multiple common vision tasks and benchmarks, despite having seen no benchmark or non-pipeline data. The depth estimation network outperforms MiDaS and the surface normal estimation network is the first to achieve human-level performance for in-the-wild surface normal estimation -- at least according to one metric on the OASIS benchmark. The Dockerized pipeline with CLI, the (mostly python) code, PyTorch dataloaders for the generated data, the generated starter dataset, download scripts and other utilities are available through our project website, https://omnidata.vision.
Key facts. Cited by 336 papers in the map (data through 7 Oct 2026). Builds on 48 of its 74 references that are in the map. First posted on arXiv 11 Oct 2021.
Builds on
48 of 74 references.
References · 48
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2014Microsoft COCO: Common Objects in Contextcited by 17,498
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2016The Cityscapes Dataset for Semantic Urban Scene Understandingcited by 4,427
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,857
- 2017CARLA: An Open Urban Driving Simulatorcited by 2,431
- 2016Matching Networks for One Shot Learningcited by 2,246
36 more
- 2021Vision Transformers for Dense Predictioncited by 1,789
- 2019Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfercited by 1,484
- 2016CLEVR: A Diagnostic Dataset for Compositional Language and Elementary Visual Reasoningcited by 1,445
- 2017Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real Worldcited by 1,401
- 2017Matterport3D: Learning from RGB-D Data in Indoor Environmentscited by 1,354
- 2019Deep High-Resolution Representation Learning for Human Pose Estimationcited by 1,339
- 2019Habitat: A Platform for Embodied AI Researchcited by 1,163
- 2017Revisiting Unreasonable Effectiveness of Data in Deep Learning Eracited by 916
- 2019The Replica Dataset: A Digital Replica of Indoor Spacescited by 885
- 2016Playing for Data: Ground Truth from Computer Gamescited by 816
- 2019Panoptic Feature Pyramid Networkscited by 644
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 631
- 2020Hypersim: A Photorealistic Synthetic Dataset for Holistic Indoor Scene Understandingcited by 593
- 2018End-to-End Multi-Task Learning with Attentioncited by 533
- 2019BlendedMVS: A Large-scale Dataset for Generalized Multi-view Stereo Networkscited by 487
- 2018Gibson Env: Real-World Perception for Embodied Agentscited by 477
- 2016Semantic Scene Completion from a Single Depth Imagecited by 468
- 2016Cross-stitch Networks for Multi-task Learningcited by 466
- 2017Joint 2D-3D-Semantic Data for Indoor Scene Understandingcited by 404
- 2018GeoNet: Unsupervised Learning of Dense Depth, Optical Flow and Camera Posecited by 341
- 20193D Scene Graph: A Structure for Unified Semantics, 3D Space, and Cameracited by 280
- 2018PAD-Net: Multi-Tasks Guided Prediction-and-Distillation Network for Simultaneous Depth Estimation and Scene Parsingcited by 224
- 2018Why do deep convolutional networks generalize so poorly to small image transformations?cited by 198
- 2016Single-Image Depth Perception in the Wildcited by 179
- 2016ViZDoom: A Doom-based AI Research Platform for Visual Reinforcement Learningcited by 178
- 2020Demystifying Contrastive Self-Supervised Learning: Invariances, Augmentations and Dataset Biasescited by 143
- 2018Continuous Learning in Single-Incremental-Task Scenarioscited by 109
- 2020Robust Learning Through Cross-Task Consistencycited by 105
- 2015Benchmarking in Manipulation Research: The YCB Object and Model Set and Benchmarking Protocolscited by 98
- 2019Branched Multi-Task Networks: Deciding What Layers To Sharecited by 76
- 2020OASIS: A Large-Scale Dataset for Single Image 3D in the Wildcited by 62
- 2019The Omniglot challenge: a 3-year progress reportcited by 54
- 2016Physically-Based Rendering for Indoor Scene Understanding Using Convolutional Neural Networkscited by 35
- 2017MirrorFlow: Exploiting Symmetries in Joint Optical Flow and Occlusion Estimationcited by 21
- 2018Left-Right Comparative Recurrent Model for Stereo Matchingcited by 12
- 2017Surface Normals in the Wildcited by 8
Built on by
Citing papers per year: 2021: 1; 2022: 20; 2023: 76; 2024: 118; 2025: 76; 2026: 45.
336 papers.
Sort:
By area:
- 2022MonoSDF: Exploring Monocular Geometric Cues for Neural Implicit Surface Reconstructioncited by 467
- 2024Metric3Dv2: A Versatile Monocular Geometric Foundation Model for Zero-shot Metric Depth and Surface Normal Estimationcited by 389
- 2024UniDepth: Universal Monocular Metric Depth Estimationcited by 375
- 2022Kubric: A scalable dataset generatorcited by 373
- 2024SV3D: Novel Multi-view Synthesis and 3D Generation from a Single Image using Latent Video Diffusioncited by 364
- 2023Metric3D: Towards Zero-shot Metric 3D Prediction from A Single Imagecited by 357
- 2025Fast3R: Towards 3D Reconstruction of 1000+ Images in One Forward Passcited by 356
- 2023OmniObject3D: Large-Vocabulary 3D Object Dataset for Realistic Perception, Reconstruction and Generationcited by 324
- 2023Repurposing Diffusion-Based Image Generators for Monocular Depth Estimationcited by 323
- 2024MoGe: Unlocking Accurate Monocular Geometry Estimation for Open-Domain Images with Optimal Training Supervisioncited by 290
- 2022MultiMAE: Multi-modal Multi-task Masked Autoencoderscited by 287
- 2024GeoWizard: Unleashing the Diffusion Priors for 3D Geometry Estimation from a Single Imagecited by 283
38 more
- 2024DepthSplat: Connecting Gaussian Splatting and Depthcited by 275
- 2023Triplane Meets Gaussian Splatting: Fast and Generalizable Single-View 3D Reconstruction with Transformerscited by 257
- 2024Sapiens: Foundation for Human Vision Modelscited by 229
- 2023OneLLM: One Framework to Align All Modalities with Languagecited by 185
- 2022ZSON: Zero-Shot Object-Goal Navigation using Multimodal Goal Embeddingscited by 177
- 2025UniDepthV2: Universal Monocular Metric Depth Estimation Made Simplercited by 170
- 2024PhysGen: Rigid-Body Physics-Grounded Image-to-Video Generationcited by 169
- 2024Lotus: Diffusion-based Visual Foundation Model for High-quality Dense Predictioncited by 163
- 2023NICER-SLAM: Neural Implicit Scene Encoding for RGB SLAMcited by 162
- 2024High-quality Surface Reconstruction using Gaussian Surfelscited by 159
- 2023DreamCraft3D: Hierarchical 3D Generation with Bootstrapped Diffusion Priorcited by 157
- 2024StableNormal: Reducing Diffusion Variance for Stable and Sharp Normalcited by 153
- 2024SV4D: Dynamic 3D Content Generation with Multi-Frame and Multi-View Consistencycited by 143
- 2024HUGS: Holistic Urban 3D Scene Understanding via Gaussian Splattingcited by 141
- 2024DN-Splatter: Depth and Normal Priors for Gaussian Splatting and Meshingcited by 136
- 2024Rethinking Inductive Biases for Surface Normal Estimationcited by 122
- 20223D Common Corruptions and Data Augmentationcited by 120
- 2023StreetSurf: Extending Multi-view Implicit Surface Reconstruction to Street Viewscited by 120
- 20234M: Massively Multimodal Masked Modelingcited by 118
- 2024MIDI: Multi-Instance Diffusion for Single Image to 3D Scene Generationcited by 113
- 2023Towards Zero-Shot Scale-Aware Monocular Depth Estimationcited by 100
- 2024Learning Temporally Consistent Video Depth from Video Diffusion Priorscited by 100
- 2025Hi3DGen: High-fidelity 3D Geometry Generation from Images via Normal Bridgingcited by 95
- 2024MVDiffusion++: A Dense High-resolution Multi-view Diffusion Model for Single or Sparse-view 3D Object Reconstructioncited by 94
- 2023PIRLNav: Pretraining with Imitation and RL Finetuning for ObjectNavcited by 91
- 2024Fine-Tuning Image-Conditional Diffusion Models is Easier than You Thinkcited by 82
- 2025Marigold: Affordable Adaptation of Diffusion-Based Image Generators for Image Analysiscited by 78
- 2024DepthFM: Fast Monocular Depth Estimation with Flow Matchingcited by 76
- 2022Offline Visual Representation Learning for Embodied Navigationcited by 74
- 2024What Matters When Repurposing Diffusion Models for General Dense Perception Tasks?cited by 74
- 2023Intrinsic Image Diffusion for Indoor Single-view Material Estimationcited by 73
- 2024How NeRFs and 3D Gaussian Splatting are Reshaping SLAM: a Surveycited by 72
- 2023OVRL-V2: A simple state-of-art baseline for ImageNav and ObjectNavcited by 68
- 2023HyperHuman: Hyper-Realistic Human Generation with Latent Structural Diffusioncited by 68
- 2024BRAVE: Broadening the visual encoding of vision-language modelscited by 68
- 2023Consistent123: One Image to Highly Consistent 3D Asset Using Case-Aware Diffusion Priorscited by 64
- 2024Splat-SLAM: Globally Optimized RGB-only SLAM with 3D Gaussianscited by 64
- 2025WorldMirror: Universal 3D World Reconstruction with Any-Prior Promptingcited by 60
50 most cited of 336 All 336 (JSON)
- 2026Fysiverse-3D-SimReady Technical Report: Agentic Physical Simulation for Pragmatic 3D World Reconstruction21 Sep 2026
- 2026Marigold V2: Revisiting Diffusion Transformers for Monocular Depth Estimation8 Sep 2026
- 2026TransNormal-2: Geometry-Grounded Rectified Flow with Edge-Aware Decoding for Precise Normal Estimation6 Sep 2026
- 2026UniFusion: Sparse-View 4D Reconstruction via Unified Spatio-temporal Depth Alignment5 Sep 2026
- 2026Monocular Depth Estimation from a Single Image: Progress and Opportunities1 Sep 2026
- 2026Efficient and High-Quality Depth Estimation via Pixel-Space Diffusion with Linear Attention31 Aug 2026
- 2026Video Generative Models as Geometry Learner28 Aug 2026
- 2026Expressivity In Multimodal Contrastive Learning17 Aug 2026
- 2026Multimodality as Supervision: Self-Supervised Specialization to the Test Environment via Multimodality16 Jul 2026
- 2026CASA-SDF: Curriculum-Aware Spatial Adaptation with Curvature-Guided Density for Neural Implicit Surface Reconstruction15 Jul 2026
- 2026ARDepth: Auto-regressive Monocular Depth Estimation with Progressive Visual Conditioning14 Jul 2026
- 2026ZipDepth: Bringing Lightweight Zero-Shot Monocular Depth Anywhere, on Any Device9 Jul 2026
38 more
- 2026From RGB Generation to Dense Field Readout: Pixel-Space Dense Prediction with Text-to-Image Models7 Jul 2026
- 2026Sparse-View Surface Reconstruction using Gaussian Splatting through High-Confidence Depth Propagation with Normal Priors4 Jul 2026
- 2026UniGP: Taming Diffusion Transformer for Prior-Preserved Unified Generation and Perception29 Jun 2026
- 2026TRON: Tracing Rays to Orchestrate a Neural Renderer for 3D Gaussian Reconstructions9 Jun 2026
- 2026TriSplat: Simulation-Ready Feed-Forward 3D Scene Reconstruction25 May 2026
- 2026Stabilizing Streaming Video Geometry via Dynamic Feature Normalization25 May 2026
- 2026GenRecon: Bridging Generative Priors for Multi-View 3D Scene Reconstruction22 May 2026
- 2026Multi-Modal Guided Multi-Source Domain Adaptation for Object Detection13 May 2026
- 2026Revisiting Photometric Ambiguity for Accurate Gaussian-Splatting Surface Reconstruction12 May 2026
- 2026The Midas Touch for Metric Depth12 May 2026
- 2026First Shape, Then Meaning: Efficient Geometry and Semantics Learning for Indoor Reconstruction5 May 2026
- 2026UniVidX: A Unified Multimodal Framework for Versatile Video Generation via Diffusion Priors1 May 2026
- 2026Monocular Depth Estimation via Neural Network with Learnable Algebraic Group and Ring Structures27 Apr 2026
- 2026HY-World 2.0: A Multi-Modal World Model for Reconstructing, Generating, and Simulating 3D Worlds15 Apr 2026
- 2026CDPR: Cross-modal Diffusion with Polarization for Reliable Monocular Depth Estimation13 Apr 2026
- 2026Generative World Renderer2 Apr 2026
- 2026Poppy: Polarization-based Plug-and-Play Guidance for Enhancing Monocular Normal Estimation29 Mar 2026
- 2026UniDAC: Universal Metric Depth Estimation for Any Camera28 Mar 2026
- 2026Unblur-SLAM: Dense Neural SLAM for Blurry Inputs26 Mar 2026
- 2026VGGT-360: Geometry-Consistent Zero-Shot Panoramic Depth Estimation19 Mar 2026
- 2026MessyKitchens: Contact-rich object-level 3D scene reconstruction17 Mar 2026
- 2026D3-RSMDE: 40× Faster and High-Fidelity Remote Sensing Monocular Depth Estimation17 Mar 2026
- 2026Iris: Bringing Real-World Priors into Diffusion Model for Monocular Depth Estimation17 Mar 2026
- 2026GeoNVS: Geometry Grounded Video Diffusion for Novel View Synthesis16 Mar 2026
- 2026PolGS++: Physically-Guided Polarimetric Gaussian Splatting for Fast Reflective Surface Reconstruction11 Mar 2026
- 2026Revisiting Shape from Polarization in the Era of Vision Foundation Models5 Mar 2026
- 2026RAD: Retrieval-Augmented Monocular Metric Depth Estimation for Underrepresented Classes10 Feb 2026
- 2026Interacted Planes Reveal 3D Line Mapping1 Feb 2026
- 2026TransNormal: Dense Visual Semantics for Diffusion-based Transparent Object Normal Estimation31 Jan 2026
- 2026Bridging Visual and Wireless Sensing via a Unified Radiation Field for 3D Radio Map Construction27 Jan 2026
- 2026NanoSD: Edge Efficient Foundation Model for Real Time Image Restoration14 Jan 2026
- 2026Pixel-Perfect Visual Geometry Estimation8 Jan 2026
- 2026AnyDepth: Depth Estimation Made Easy6 Jan 2026
- 2025Visual Autoregressive Modelling for Monocular Depth Estimation27 Dec 2025
- 2025MatE: Material Extraction from Single-Image via Geometric Prior20 Dec 2025
- 2025Accuracy Does Not Guarantee Human-Likeness: Cross-Domain Human-Centered Benchmark in Monocular Depth Estimation9 Dec 2025
- 2025D3-Predictor: Noise-Free Deterministic Diffusion for Dense Prediction8 Dec 2025
- 2025Lotus-2: Advancing Geometric Dense Prediction with Powerful Image Generative Model30 Nov 2025
50 newest of 336 All 336 (JSON)
- 2024SV3D: Novel Multi-view Synthesis and 3D Generation from a Single Image using Latent Video Diffusioncited by 364
- 2023Repurposing Diffusion-Based Image Generators for Monocular Depth Estimationcited by 323
- 2024GeoWizard: Unleashing the Diffusion Priors for 3D Geometry Estimation from a Single Imagecited by 283
- 2024Lotus: Diffusion-based Visual Foundation Model for High-quality Dense Predictioncited by 163
- 2023DreamCraft3D: Hierarchical 3D Generation with Bootstrapped Diffusion Priorcited by 157
- 2024StableNormal: Reducing Diffusion Variance for Stable and Sharp Normalcited by 153
- 2024SV4D: Dynamic 3D Content Generation with Multi-Frame and Multi-View Consistencycited by 143
- 2024MIDI: Multi-Instance Diffusion for Single Image to 3D Scene Generationcited by 113
- 2024Learning Temporally Consistent Video Depth from Video Diffusion Priorscited by 100
- 2024MVDiffusion++: A Dense High-resolution Multi-view Diffusion Model for Single or Sparse-view 3D Object Reconstructioncited by 94
- 2024Fine-Tuning Image-Conditional Diffusion Models is Easier than You Thinkcited by 82
- 2025Marigold: Affordable Adaptation of Diffusion-Based Image Generators for Image Analysiscited by 78
12 most cited of 61 All 61 (JSON)
- 2026Marigold V2: Revisiting Diffusion Transformers for Monocular Depth Estimation8 Sep 2026
- 2026TransNormal-2: Geometry-Grounded Rectified Flow with Edge-Aware Decoding for Precise Normal Estimation6 Sep 2026
- 2026Video Generative Models as Geometry Learner28 Aug 2026
- 2026ARDepth: Auto-regressive Monocular Depth Estimation with Progressive Visual Conditioning14 Jul 2026
- 2026UniGP: Taming Diffusion Transformer for Prior-Preserved Unified Generation and Perception29 Jun 2026
- 2026UniVidX: A Unified Multimodal Framework for Versatile Video Generation via Diffusion Priors1 May 2026
- 2026D3-RSMDE: 40× Faster and High-Fidelity Remote Sensing Monocular Depth Estimation17 Mar 2026
- 2026Iris: Bringing Real-World Priors into Diffusion Model for Monocular Depth Estimation17 Mar 2026
- 2026GeoNVS: Geometry Grounded Video Diffusion for Novel View Synthesis16 Mar 2026
- 2026NanoSD: Edge Efficient Foundation Model for Real Time Image Restoration14 Jan 2026
- 2025D3-Predictor: Noise-Free Deterministic Diffusion for Dense Prediction8 Dec 2025
- 2025Lotus-2: Advancing Geometric Dense Prediction with Powerful Image Generative Model30 Nov 2025
12 newest of 61 All 61 (JSON)
- 2023OneLLM: One Framework to Align All Modalities with Languagecited by 185
- 2024BRAVE: Broadening the visual encoding of vision-language modelscited by 68
- 2026Expressivity In Multimodal Contrastive Learningcited by 0
- 2026Expressivity In Multimodal Contrastive Learning17 Aug 2026
- 2024BRAVE: Broadening the visual encoding of vision-language models10 Apr 2024
- 2023OneLLM: One Framework to Align All Modalities with Language6 Dec 2023
- 2022MonoSDF: Exploring Monocular Geometric Cues for Neural Implicit Surface Reconstructioncited by 467
- 2024Metric3Dv2: A Versatile Monocular Geometric Foundation Model for Zero-shot Metric Depth and Surface Normal Estimationcited by 389
- 2024UniDepth: Universal Monocular Metric Depth Estimationcited by 375
- 2022Kubric: A scalable dataset generatorcited by 373
- 2023Metric3D: Towards Zero-shot Metric 3D Prediction from A Single Imagecited by 357
- 2025Fast3R: Towards 3D Reconstruction of 1000+ Images in One Forward Passcited by 356
- 2023OmniObject3D: Large-Vocabulary 3D Object Dataset for Realistic Perception, Reconstruction and Generationcited by 324
- 2024MoGe: Unlocking Accurate Monocular Geometry Estimation for Open-Domain Images with Optimal Training Supervisioncited by 290
- 2022MultiMAE: Multi-modal Multi-task Masked Autoencoderscited by 287
- 2024DepthSplat: Connecting Gaussian Splatting and Depthcited by 275
- 2023Triplane Meets Gaussian Splatting: Fast and Generalizable Single-View 3D Reconstruction with Transformerscited by 257
- 2024Sapiens: Foundation for Human Vision Modelscited by 229
12 most cited of 268 All 268 (JSON)
- 2026Fysiverse-3D-SimReady Technical Report: Agentic Physical Simulation for Pragmatic 3D World Reconstruction21 Sep 2026
- 2026UniFusion: Sparse-View 4D Reconstruction via Unified Spatio-temporal Depth Alignment5 Sep 2026
- 2026Monocular Depth Estimation from a Single Image: Progress and Opportunities1 Sep 2026
- 2026Efficient and High-Quality Depth Estimation via Pixel-Space Diffusion with Linear Attention31 Aug 2026
- 2026Multimodality as Supervision: Self-Supervised Specialization to the Test Environment via Multimodality16 Jul 2026
- 2026CASA-SDF: Curriculum-Aware Spatial Adaptation with Curvature-Guided Density for Neural Implicit Surface Reconstruction15 Jul 2026
- 2026ZipDepth: Bringing Lightweight Zero-Shot Monocular Depth Anywhere, on Any Device9 Jul 2026
- 2026From RGB Generation to Dense Field Readout: Pixel-Space Dense Prediction with Text-to-Image Models7 Jul 2026
- 2026Sparse-View Surface Reconstruction using Gaussian Splatting through High-Confidence Depth Propagation with Normal Priors4 Jul 2026
- 2026TRON: Tracing Rays to Orchestrate a Neural Renderer for 3D Gaussian Reconstructions9 Jun 2026
- 2026TriSplat: Simulation-Ready Feed-Forward 3D Scene Reconstruction25 May 2026
- 2026Stabilizing Streaming Video Geometry via Dynamic Feature Normalization25 May 2026
12 newest of 268 All 268 (JSON)