Yansheng Li |
Junwei Luo* |
Yongjun Zhang* |
Yihua Tan |
Jin-Gang Yu |
Song Bai
* Indicates corresponding authors
- Dataset@Baidu Drive: Link
- Dataset@Hugging Face: Link
- Evaluation: Codabench Evaluation
- Weight: HBD-Net
Bridges represent critical infrastructure components, serving as fundamental transportation facilities that traverse various landscapes. They hold substantial significance in the domains like civil transportation and disaster relief efforts. To ensure the visibility and integrity of bridges, it is essential to perform holistic bridge detection in large-size very-high-resolution (VHR) remote sensing images (RSIs).
However, large-scale datasets and methods suitable for holistic bridge detection in large-sisze RSIs have yet to be explored. This work aims to addresses the challenge of holistic bridge detection in large-size VHR RSIs. Key contributions include:
- GLH-Bridge: A large-scale dataset with 6,000 globally sampled VHR images (from 2,048 x 2,048 to 16,384 x 16,384 pixels) and 59,737 manually annotated bridges.
- HBD-Net: A novel method for holistic bridge detection.
- Comprehensive benchmarks for both Oriented Bounding Box (OBB) and Horizontal Bounding Box (HBB) tasks.
🌍 Diverse Coverage
6,000 images spanning global geographic locations with varied backgrounds (vegetation, riverbeds, roads).
🖼️ Very-High-Resolution
Image sizes range from 2,048×2,048 to 16,384×16,384 pixels.
📊 Rich Annotations
59,737 bridges annotated with both OBB and HBB labels, featuring:
- Extreme aspect ratios
- Multi-scale instances
- Varied instance densities
Please employ the following scripts for training and inference:
-
Training:
- Multi-resolution data splitting:
tools/split_code.py
- Training script:
train_dist_mmrot.sh
- Load checkpoint for stage-by-stage training:
tools/loadckpt_backbone.py
- Multi-resolution data splitting:
-
Inference:
- Test script:
test_mmrot_all.sh
- Test script:
If you find this work helpful for your research, please consider citing our papers:
@article{li2024learning,
title={Learning to Holistically Detect Bridges From Large-Size VHR Remote Sensing Imagery},
author={Li, Yansheng and Luo, Junwei and Zhang, Yongjun and Tan, Yihua and Yu, Jin-Gang and Bai, Song},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
volume={44},
number={11},
pages={7778--7796},
year={2024},
publisher={IEEE}
}
@ARTICLE{STAR,
title={STAR: A First-Ever Dataset and a Large-Scale Benchmark for Scene Graph Generation in Large-Size Satellite Imagery},
author={Li, Yansheng and Wang, Linlin and Wang, Tingzhu and Yang, Xue and Luo, Junwei and Wang, Qi and Deng, Youming and Wang, Wenbin and Sun, Xian and Li, Haifeng and Dang, Bo and Zhang, Yongjun and Yu, Yi and Yan, Junchi},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
year={2025},
volume={47},
number={3},
pages={1832-1849},
doi={10.1109/TPAMI.2024.3508072},
publisher={IEEE}
}