Skip to content

Implementation of "Bayesian MEF" as referred in the publication "Bayesian multi-exposure image fusion for robust high dynamic range ptychography"

License

Notifications You must be signed in to change notification settings

microscopic-image-analysis/bayes-mef

Repository files navigation

Bayesian MEF

PyPI Python 3.9+ DOI License

Bayesian multi-exposure image fusion (MEF) is a general-purpose algorithm to achieve robust high dynamic range (HDR) imaging, particularly in scenarios with low signal-to-noise ratio (SNR) or variations in illumination intensity. This approach could especially be useful for high quality phase retrieval in coherent diffractive imaging (CDI). The algorithm, detailed in this publication "Bayesian multi-exposure image fusion for robust high dynamic range ptychography", explains the method and demonstrates the benefits for ptychography. To reproduce the results in the paper, see this section. However, to get started, please install the package and check the demo usage below.

demo_mef

To install this package from PyPI,

pip install bayes_mef

Usage

A minimal example demonstrating the usage of BayesianMEF by simulating some data.

Open In Colab

from bayes_mef import BayesianMEF
from skimage.data import camera
import numpy as np

# simulation params
truth = camera()
background = 60  # some background
times = np.array([0.1, 1, 10])  # exposure times or equivalently flux factors
threshold = 1500 # detector limit

# poisson data based on image formation model that is overexposed
data = [np.random.poisson(time * truth + background) for time in times]
data_saturated = np.clip(data, None, threshold, dtype="float")

# Bayesian MEF with optional field `update_fluxes`. Set it to `True` when
# flux factors (exposure times) are not accurately known.
mef_em = BayesianMEF(data_saturated, threshold, times, background, update_fluxes=False)
mef_em.run(n_iter=100)
fused_em = mef_em.fused_image.copy()

Additionally, one can also use the ConventionalMEF method as given in the paper.

from bayes_mef import ConventionalMEF
mef_mle = ConventionalMEF(data_saturated, threshold, times, background)
mef_mle.mle()
fused_mle = mef_mle.fused_image.copy()

Parallelized implementatation

In ptychography, one needs to fuse diffraction patterns for every scan position. Processing this for data over all scan positions can be slow for an iterative algorithm. Therefore, one can use the parallelized implementation for MEF LaunchMEF. Check the example below

from bayes_mef import LaunchMEF

launch_mef = LaunchMEF(
    ptychogram_stack,    # ptychogram shape (n_exposures, n_scans, dp_x, dp_y)
    background,          # background shape (n_exposures, n_scans, dp_x, dp_y)
    flux_factors=None,   # if set to `None`, calculates automatically
    threshold=None,      # if set to `None`, calculates automatically
    update_fluxes=False, # set to `True` if you want to update fluxes
)

# runs Bayesian MEF in parallel by defining the number of CPUs `n_cpus`;
# returns the fused diffraction patterns with shape (n_scans, dp_x, dp_y) and updated flux factors
n_cpus = 20
n_iter = 150
fused_ptyem_stack, em_flux_factors = launch_mef.run_em(n_iter, n_cpus)

For a detailed usage, please check synthetic_mef.py that uses synthetic ptychography data.

Reproducing results

To reproduce the ptychographic reconstruction results from the paper, please follow the below steps:

  1. Please clone this repository and create a conda environment.

    git clone https://github.com/microscopic-image-analysis/bayes-mef.git
    cd bayes-mef
    conda create --name bayes-mef-venv python=3.11.5 
  2. Now activate the environment and install the pinned dependencies.

    conda activate bayes-mef-venv
    pip install -r requirements.txt
  3. Download the data from Zenodo with the following command:

    ./download_data.sh
  4. Optional: For faster ptychographic reconstructions using GPU, please install cupy as given under its installation guide.

  5. Run files from the scripts/ directory for plotting the results.

Citation

If you found this algorithm or the publication useful, please cite us at:

@article{Kodgirwar:24,
author = {Shantanu Kodgirwar and Lars Loetgering and Chang Liu and Aleena Joseph and Leona Licht and Daniel S. Penagos Molina and Wilhelm Eschen and Jan Rothhardt and Michael Habeck},
journal = {Opt. Express},
number = {16},
pages = {28090--28099},
publisher = {Optica Publishing Group},
title = {Bayesian multi-exposure image fusion for robust high dynamic range ptychography},
volume = {32},
month = {Jul},
year = {2024},
url = {https://opg.optica.org/oe/abstract.cfm?URI=oe-32-16-28090},
doi = {10.1364/OE.524284},
}

About

Implementation of "Bayesian MEF" as referred in the publication "Bayesian multi-exposure image fusion for robust high dynamic range ptychography"

Resources

License

Stars

Watchers

Forks

Packages

No packages published