SPM / README.md
prasannareddyp's picture
Update README.md
a8a94da verified
---
title: Shuffle PatchMix
colorFrom: purple
colorTo: red
sdk: gradio
sdk_version: 5.44.1
app_file: app.py
pinned: false
license: mit
emoji: 🔥
short_description: An interactive demo for SPM augmentation.
---
# Shuffle PatchMix (SPM) — Hugging Face Space
An interactive demo for SPM augmentation.
GitHub repo: https://github.com/PrasannaPulakurthi/SPM
arxiv.org/abs/2505.24216
## Parameters
- **Grid (N×N):** Choose one of **2×2, 4×4, 8×8, 16×16**. The image is cropped (top-left) so its width and height are divisible by N.
- **Enable Overlap Patch Blend):** When enabled, each base cell expands by ±overlap pixels (2× at borders to keep areas comparable) and patches are blended with a feather mask.
- **Overlap (px):** Pixel overlap per side. Automatically clipped to `< 1/2 * patch size`.
- **Mix probability:** Per-patch probability to mix original and a shuffled patch.
- **Beta α, β:** Shape parameters for the Beta distribution used for blending weights.
- **Seed:** Optional deterministic seed.
## Batch Mode
Upload a `.zip` containing images (`.png`, `.jpg`, `.jpeg`). The app returns a `.zip` of augmented results with the same folder structure.
## Local Development
```bash
pip install -r requirements.txt
python app.py
```
## License
MIT