README.md

---
title: Dual-Region Augmentation (DRA, Local U²-Net)
colorFrom: indigo
colorTo: blue
sdk: gradio
sdk_version: 5.44.1
app_file: app.py
pinned: false
license: mit
short_description: An interactive demo for DRA augmentation.
---

### Dual-Region Foreground-Background Augmentation (DRA)

An interactive demo for DRA augmentation.

GitHub repo: https://github.com/PrasannaPulakurthi/Foreground-Background-Augmentation

arxiv.org/abs/2504.13077

- **Grid control**: choose number of patches (2×2, 4×4, 8×8, 16×16). No overlap.
- **Background**: jigsaw shuffle on the full image, fused only into background via U²‑Net mask.
- **Foreground**: Gaussian noise applied to **one random rectangular box** inside the foreground.
- **Weights**: uses local `saved_models/u2net(.pth)` or `saved_models/u2netp(.pth)`.