Skip to content

CoInfra: A Large-Scale Cooperative Infrastructure Perception System and Dataset in Adverse Weather

License

Notifications You must be signed in to change notification settings

NingMingHao/CoInfra

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CoInfra: A Large-Scale Cooperative Infrastructure Perception System and Dataset in Adverse Weather

📄 Arxiv Paper

📦 Dataset Download

📂 Codebase Overview

🏗️ Build Your Own Cooperative Perception System

CoInfra is a real-world, large-scale cooperative perception system and dataset designed to support research in multi-agent perception, especially under adverse weather conditions (snow, rain, freezing rain). It features 14 synchronized infrastructure nodes, each with a LiDAR and dual cameras, deployed across a shared region for full-scene coverage.

We release the dataset and baseline experiments to facilitate research in:

  • Multi-node 3D object detection and tracking in adverse weather
  • Delay-aware synchronization for real-time perception
  • Motion prediction in interaction-rich roundabout scenario

The codebase includes:

  • Dataset visualization tools for 2D and 3D
  • Web-based UI for mass device management, OTA updates, and a real-time visualization, etc if you are interested in 🏗️ Build Your Own Cooperative Perception System
  • Baseline models for early and late fusion
  • A ROS-based Sensor Node level and Cloud level late fusion system

📦 Dataset Download

The CoInfra dataset includes around:

  • 195k LiDAR frames
  • 390k camera images
  • 220k annotated 3D bounding boxes across 5 classes (car, bus, truck, bicycle, pedestrian)
  • Adverse weather scenes: ☀️ Sunny, 🌧️ Rainy, ❄️ Snowy, 🧊 Freezing Rain
  • HD map
  • Calibration data for camera-LiDAR and global alignment

🔗 Download Links

  • A mini example dataset (1 slice, 1 scenario) is available for quick testing: Mini Example Dataset
  • Email to [email protected] for requesting the data
  • Download the dataset from HuggingFace: Static Badge

Refer to docs/DATASET.md for data format, structure, and annotation schema.


📂 Codebase Structure

1. Environment Setup

conda create -n coinfra python=3.10
conda activate coinfra
pip install -r requirements.txt

2. Visualization Tools

  • 2D Visualization: visualization/2D/main.py

    • You need to set the BASE_FOLDER_PATH variable to the path of your dataset folder.
    • Then run python visualization/2D/main.py to visualize the dataset in a web-based UI.
  • 3D Visualization: visualization/3D/main.py

    • Run python visualization/3D/main.py to visualize the dataset in a 3D viewer.
    • Click Load Scenario to go to the scenario/slice which you want to visualize, and then Open to load the data.

🏗️ Build Your Own Cooperative Perception System

Hardware

We provide a reference hardware setup for building your own cooperative perception system:

  • LiDAR: Robosense Helios 1615
  • Cameras: Two Basler dart daA1920-160uc global shutter RGB cameras
  • Computing Unit: NVIDIA Jetson Orin NX 16GB
  • Power System: Renogy Wanderer 10A solar charge controller with a dedicated solar panel and battery
  • Communication Module: Industrial-grade 5G modem for real-time data transmission. The modem supports secure remote access and management, facilitating OTA updates and system diagnostics. Integrated GPIO and serial interfaces enable advanced IoT functionalities, including remote power control via relay modules and continuous monitoring of the solar power system through serial communication with the charge controller

Software

  • Web-based UI: A web interface for managing multiple nodes, performing OTA updates, and visualizing real-time data.
UIDemo.mp4
  • ROS-based Sensor Node:
SensorNodePerception.mp4
  • ROS-based Cloud:
CloudPerception.mp4

More detailed documentation is in docs/SETUP.md

🔍 Baseline Experiments

We provide benchmarks for early and late fusion using BEV projections:

Fusion Strategy HD Map mAP (All Classes)
Early Fusion 0.984
Early Fusion ✔️ 0.986
Late Fusion 0.931
Late Fusion ✔️ 0.952

📖 Citation

If you use CoInfra, please cite:

@article{ning2025coinfra,
  title={CoInfra: A Large-Scale Cooperative Infrastructure Perception System and Dataset in Adverse Weather},
  author={Ning, Minghao and Yang, Yufeng and Shu, Keqi and Huang, Shucheng and Zhong, Jiaming and Salehi, Maryam and Rahmani, Mahdi and Lu, Yukun and Sun, Chen and Saleh, Aladdin and Hashemi, Ehsan and Khajepour, Amir},
  journal={arXiv preprint arXiv:2507.02245},
  url={https://arxiv.org/abs/2507.02245},
  year={2025}
}

📬 Contact

Questions or issues? Please open an issue or contact [email protected].

📝 License

This project is licensed under the CC BY-NC 4.0 License - see the LICENSE file for details.

About

CoInfra: A Large-Scale Cooperative Infrastructure Perception System and Dataset in Adverse Weather

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages