pip install git+https://github.com/sswlab/ssw-tools

Raw AIA Level 1 FITS (4096x4096, variable orientation/brightness)
    │
    ├─ 1. Pointing Correction   → Fix spacecraft orientation errors
    ├─ 2. Registration           → North-up, center disk, resample to fixed size
    ├─ 3. Degradation Correction → Compensate sensor aging over mission lifetime
    └─ 4. Exposure Normalization → DN → DN/s (standardize brightness)
    │
    ▼
ML-Ready (1024x1024, float32, DN/s, north-up, centered)

from sunpy.map import Map
import astropy.units as u
from aiapy.calibrate.util import get_correction_table, get_pointing_table
from ssw_tools.prep.sdo_aia import aia_prep_ml

# Load raw FITS
aia_map = Map('aia_lev1_file.fits')

# Fetch calibration tables
pointing_table = get_pointing_table(aia_map.date - 6*u.h, aia_map.date + 6*u.h)
correction_table = get_correction_table()

# Preprocess
prep_map = aia_prep_ml(
    aia_map,
    pointing_table=pointing_table,
    correction_table=correction_table,
    resolution=1024,        # output size: 512, 1024, or 2048
    padding_factor=0.1      # solar disk padding: 0.1=10%, 0.2=20%
)

# Save
prep_map.save('prep_193A.fits', overwrite=True)

from pathlib import Path
from sunpy.map import Map
import astropy.units as u
from aiapy.calibrate.util import get_correction_table, get_pointing_table
from ssw_tools.prep.sdo_aia import aia_prep_ml
from loguru import logger

input_dir = Path('./raw/')
output_dir = Path('./prep/')
output_dir.mkdir(exist_ok=True)

correction_table = get_correction_table()  # fetch once, reuse

for fits_file in sorted(input_dir.glob('*.fits')):
    try:
        m = Map(str(fits_file))
        pt = get_pointing_table(m.date - 6*u.h, m.date + 6*u.h)
        prep = aia_prep_ml(m, pt, correction_table, resolution=1024, padding_factor=0.1)
        prep.save(str(output_dir / f'prep_{m.wavelength.value:.0f}A_{m.date.isot}.fits'),
                  overwrite=True)
        logger.info(f'Done: {fits_file.name}')
    except Exception as e:
        logger.error(f'Failed: {fits_file.name}: {e}')