DeepMIMO 4.0.0

DeepMIMO dataset generator library

DeepMIMO

Bridging ray tracers and 5G/6G simulators with shareable, site-specific datasets

Quickstart • Docs • Web • FAQ • Cite

G O A L	H O W	W H Y
Enable large‑scale AI benchmarking using site‑specific wireless ray‑tracing datasets.	Convert outputs from top propagation ray tracers into a clean, distributable format readable by modern simulation toolboxes.	Make ray‑tracing data easy to access, share, and benchmark—accelerating AI‑driven wireless research.

Features

• ⚡ Instant access to ray-tracing data — Compute large, realistic channels in seconds.
• 🧪 Easy ML benchmarking across sites — Find 100s of datasets in the Scenarios Database
• 🔁 Reproduce benchmarks — Search papers by topic and application in Publications Database.
• 🚀 Feature-rich toolbox — Explore a wide array of wireless utilities in our Notebook Tutorials.
• 🔌 Seamless integration — From Sionna RT/InSite/AODT to Sionna/MATLAB 5G/NeoRadium.
• 📦 Shareable datasets — Versioned scenarios, open formats. Explore in the Online Visualizer
• 🤗 User friendly — Great docs, practical examples, easy install, and available on Colab.
• 🌍 Active Community & Support — Issues and Pull Requests reviewed in hours not weeks.

Quickstart

Install

# From PyPI
pip install deepmimo

# From GitHub
git clone https://github.com/DeepMIMO/DeepMIMO.git
cd DeepMIMO
pip install -e .[dev]

Basic Dataset Generation

import deepmimo as dm

# Download a dataset
dm.download('asu_campus_3p5')

# Load a dataset
dataset = dm.load('asu_campus_3p5')

# Generate channels
channels = dataset.compute_channels()  # [n_ue, n_rx, n_tx, n_sub]

Convert Ray Tracing Simulations to DeepMIMO

import deepmimo as dm

# Convert Wireless Insite, Sionna, or AODT to DeepMIMO
scenario_name = dm.convert('path_to_ray_tracing_output')

# Upload a dataset to the DeepMIMO Database (optional)
dm.upload(scenario_name, 'api-key')
# get api key in deepmimo.net -> contribute

Project Structure

deepmimo/
├── api/                    # Database API
│   ├── download.py         # Download scenarios from database
│   ├── search.py           # Search scenarios in database
│   └── upload.py           # Upload scenarios to database
├── converters/             # Ray tracer output converters
│   ├── aodt/               # AODT converter
│   ├── sionna_rt/          # Sionna RT converter
│   ├── wireless_insite/    # Wireless Insite converter
│   ├── converter.py        # Base converter class
│   └── converter_utils.py  # Converter utilities
├── core/                   # Core data models
│   ├── materials.py        # Material properties
│   ├── rt_params.py        # Ray tracing parameters
│   ├── scene.py            # Physical environment representation
│   └── txrx.py             # Transmitter/receiver configurations
├── datasets/               # Dataset operations
│   ├── array_wrapper.py    # Array management utilities
│   ├── dataset.py          # Dataset, MacroDataset, DynamicDataset classes
│   ├── generate.py         # Dataset generation with channel computation
│   ├── load.py             # Dataset loading functionality
│   ├── sampling.py         # User sampling utilities
│   ├── summary.py          # Dataset summary functions
│   └── visualization.py    # Plotting and visualization tools
├── exporters/              # Data exporters
│   ├── aodt_exporter.py    # AODT format exporter
│   └── sionna_exporter.py  # Sionna format exporter
├── generator/              # Channel generation
│   ├── ant_patterns.py     # Antenna pattern definitions
│   ├── channel.py          # MIMO channel computation
│   └── geometry.py         # Geometric calculations and beamforming
├── integrations/           # Integration with 5G simulation tools
│   ├── matlab/             # MATLAB 5G Toolbox integration
│   ├── sionna_adapter.py   # Sionna integration
│   └── web.py              # DeepMIMO web format export
├── pipelines/              # Automatic ray tracing pipelines
│   ├── sionna_rt/          # Sionna raytracer pipeline
│   ├── wireless_insite/    # Wireless Insite pipeline
│   ├── blender_osm.py      # Blender OSM export utilities
│   ├── txrx_placement.py   # Transmitter/receiver placement
│   └── utils/              # Pipeline utilities
├── utils/                  # Utility modules
│   ├── data_structures.py  # Custom data structures
│   ├── dict_utils.py       # Dictionary utilities
│   ├── geometry.py         # Geometric utility functions
│   ├── info.py             # Information on matrices and parameters
│   ├── io.py               # File I/O operations
│   └── scenarios.py        # Scenario management functions
├── config.py               # Configuration management
└── consts.py               # Constants and default values

Additional directories:
├── docs/                   # Documentation
├── scripts/                # Utility scripts
└── tests/                  # Test suite

Build the Docs

After cloning the repository:

Step	Command	Description
Install	pip install .[dev]	Install development dependencies
Serve	mkdocs serve	Preview at http://localhost:8000

Change execute: false to execute: true in mkdocs.yml to run the tutorials and preserve cell outputs.

Contributing

We welcome contributions to DeepMIMO! To contribute:

1. Fork the repository
2. Make changes
3. Open a Pull Request

We aim to respond to pull requests within 24 hours.

FAQ

1) What is DeepMIMO useful for?

• Free, easy and fast access to ray tracing data across hundreds of site-specific datasets.
• Connecting raytracers and simulators, allowing flexible research.
• Sharing Datasets to make research more easily reproducible

2) Why using DeepMIMO if Sionna exists?

DeepMIMO is not a simulator; it’s a standardized ray-tracing toolchain that relies and complements ray-tracing tools like Sionna RT. Re-running a single high-quality scenario can take hours or days of compute, provided one can resurrect the original scripts and software environment. And even when this succeeds, every group tends to store the channels in its own ad-hoc format, so results are not plug-and-play across projects. With DeepMIMO, we skip all that: authors publish a scenario once, and anyone can load the exact same data in seconds with a small Python snippet. DeepMIMO makes sharing, reproducing, and comparing ray-tracing results dramatically easier and more reliable.

3) How long do dataset downloads take?

A few minutes. Sometimes seconds. Data is stored on super-fast and available object storage, so in practice the transfer speeds are limited by the internet connection (see speedtest.net). At 50 Mbps the ASU Campus scenario with 85 thousand candidate users would take ~5 seconds to download.

Citation

If you use this software, please cite it:

@misc{alkhateeb2019deepmimo,
      title={DeepMIMO: A Generic Deep Learning Dataset for Millimeter Wave and Massive MIMO Applications}, 
      author={Ahmed Alkhateeb},
      year={2019},
      eprint={1902.06435},
      archivePrefix={arXiv},
      primaryClass={cs.IT},
      url={https://arxiv.org/abs/1902.06435}, 
}