RSGen: Enhancing Layout-Driven Remote Sensing Image Generation with Diverse Edge Guidance
Organizations: School of Future Science and Engineering, Soochow University, Suzhou, China · D-Robotics, Beijing, China · Fraunhofer Institute for Applied Information Technology, Sankt Augustin, Germany · School of Mathematical and Computing Sciences, Heriot-Watt University Malaysia, Putrajaya, Malaysia ·
Abstract
Diffusion models have significantly mitigated the impact of annotated data scarcity in remote sensing (RS). Although recent approaches have successfully harnessed these models to enable diverse and controllable Layout-to-Image (L2I) synthesis, they still suffer from limited fine-grained control and fail to strictly adhere to bounding box constraints. To address these limitations, we propose RSGen, a plug-and-play framework that leverages diverse edge guidance to enhance layout-driven RS image generation. Specifically, RSGen employs a progressive enhancement strategy: 1) it first enriches the diversity of edge maps composited from retrieved training instances via Image-to-Image generation; and 2) subsequently utilizes these diverse edge maps as conditioning for existing L2I models to enforce pixel-level control within bounding boxes, ensuring the generated instances strictly adhere to the layout. Extensive experiments across three baseline models demonstrate that RSGen significantly boosts the capabilities of existing L2I models. For instance, with CC-Diff on the DOTA dataset for oriented object detection, we achieve remarkable gains of +9.8/+12.0 in YOLOScore mAP50/mAP50-95 and +1.6 in mAP on the downstream detection task. Our code will be publicly available: https://github.com/D-Robotics-AI-Lab/RSGen
Figures & tables
| Method | DIOR-RSVG (HBB) | DOTA (OBB) | |||||
|---|---|---|---|---|---|---|---|
| FID | YOLOScore | FID | YOLOScore | ||||
| MIGC [ 38 ] | 79.32 | 63.2 | 38.4 | 66.80 | 53.0 | 30.0 | |
| + Ours | 85.04 | 68.7 | 45.8 | 68.07 | 55.1 | 37.7 | |
| CC-Diff [ 35 ] | 66.75 | 66.8 | 41.2 | 47.72 | 57.7 | 34.0 | |
| + Ours | 68.12 | 70.1 | 47.7 | 46.86 | 67.5 | 46.0 | |
| Method | DIOR-RSVG (HBB) | DOTA (HBB) | DOTA (OBB) | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
| mAP | mAP | |||||||||
| MIGC [ 38 ] | 54.3 | 78.9 | 59.8 | 38.3 | 64.5 | 39.2 | 53.55 | 19.68 | ||
| + Ours | 55.0 | 78.7 | 61.1 | 38.9 | 64.5 | 40.5 | 53.19 | 21.92 | ||
| CC-Diff [ 35 ] | 54.7 | 78.4 | 60.1 | 37.4 | 63.2 | 38.6 | 54.80 | 20.11 | ||
| + Ours | 55.0 | 78.8 | 60.7 | 39.0 | 64.2 | 40.4 | 54.92 | 21.65 | ||
| FICGen [ 24 ] | 54.5 | 78.7 | 60.7 | 39.2 | 64.8 | 40.8 | 55.81 | 22.66 | ||
| Method | Ratio (Real:Syn) | DOTA (HBB) | DOTA (OBB) | ||||
|---|---|---|---|---|---|---|---|
| mAP | |||||||
| CC-Diff [ 35 ] | 1:3 | 39.0 | 65.2 | 39.8 | 55.72 | 21.11 | |
| + Ours | 39.0 | 64.2 | 40.0 | 56.07 | 24.36 | ||
| CC-Diff [ 35 ] | 1:2 | 38.8 | 65.4 | 39.7 | 54.75 | 21.44 | |
| + Ours | 39.3 | 64.4 | 41.2 | 55.98 | 22.34 | ||
| CC-Diff [ 35 ] | 1:1 | 37.4 | 63.2 | 38.6 | 54.80 | 20.11 | |
| Method | FID | YOLOScore | |
|---|---|---|---|
| w/o Edge2Edge | 68.40 | 69.5 | 46.9 |
| w/ Edge2Edge | 68.12 | 70.1 | 47.7 |
| Method | vehicle | ship | basketballcourt | groundtrackfield | harbor | tenniscourt | airplane |
|---|---|---|---|---|---|---|---|
| w/o Edge2Edge | 6.8 | 8.3 | 14.9 | 13.6 | 9.6 | 42.6 | 8.2 |
| w/ Edge2Edge | 7.1 | 8.3 | 15.5 | 13.5 | 10.2 | 44.6 | 8.4 |
| Method | FID | YOLOScore | |
|---|---|---|---|
| Baseline (Global Control) | 112.43 | 62.1 | 45.5 |
| + Spatially Gated | 64.38 | 68.7 | 45.7 |
| + High-pass Filter (Full) | 68.40 | 69.5 | 46.9 |
Image Generation
Image generation studies models that create or edit images from text, examples, layouts, and other structured controls.
See all 463 papers