Automatically stretch linear astronomical images for easy visualization.
Auto Stretch applies a PixInsight-style Screen Transfer Function (STF) midtones transfer to linear astronomical images stored as numpy arrays. It automatically computes the optimal stretch parameters from the image data so that faint detail becomes visible without manual histogram adjustment.
| linear (no stretch) | arcsin stretch | auto-stretch (using this library) |
|---|---|---|
 |
 |
 |
pip install auto-stretch
import numpy as np
from auto_stretch import apply_stretch, Stretch
image = np.array([[0.1, 0.25], [0.8, 0.3]], dtype=np.float64)
stretched = apply_stretch(image)
# or with custom parameters:
stretched = Stretch(target_bkg=0.25, shadows_clip=-1.25).stretch(image)- Input must be a 2D numpy array. 1D and 3D arrays raise
ValueError. - Input is coerced to float64 internally; the caller's array is never mutated.
- NaN pixels propagate through the stretch unchanged.
- Output is float64 with values clamped to
[0, 1].
| Parameter | Default | Description |
|---|---|---|
target_bkg |
0.25 | Target median brightness after stretch |
shadows_clip |
-1.25 | Shadow clipping point in units of median absolute deviation below the median |
The underlying algorithm follows the PixInsight Histogram Transformation STF method.
- Python >= 3.10
- numpy >= 1.20
Based on the PixInsight Screen Transfer Function (STF) algorithm developed by Pleiades Astrophoto.
MIT — see LICENSE.