Python implementation of computational optical propagation and digital holography.
Project description
Fringe.Py
A coherent propagation and diffraction simulation tool, all written in Python.
In this package, a set of utilities are provided to simulate coherent signal propagation. It is particularly made for free-space optical propagation, diffraction, and holography. However, the tools are compatible with 1D and 2D data structures and can be potentially used for any sort of spatially-coherent signals.
What's inside?
A set of modules are provided to solve the scalar diffraction problem. The codes are GPU-friendly, compatible with well-known computation libraries (Numpy, TensorFlow) and they support batch processing. Using the TensorFlow backend, computations become autograd-compatible and could be mounted on machine learning models. Angular spectrum algorithm is the primary work horse for field propagation though other custom solvers could be used. Aside the built-in Numpy and TensorFlow backends, any computational means could be employed to process tensor operations.
It also includes:
- a simple yet useful data pipeline to load and standardize data. For now, it only supports images.
- Gerchberg-Saxton multi-distance phase recovery algorithm. It can be easily tweaked to support other variations of signal e.g. by wavelength.
Installation
To install the package, run:
python -m pip install fringe
Fringe requires numpy
, tensorflow 2.x
, and scikit_image
.
The example files are not included in the package and should be downloaded separately. Also they require matplotlib
to show plots.
How to Use
- Import or create data
For images:
import numpy as np
from fringe.utils.io import import_image
from fringe.utils.modifiers import ImageToArray, Normalize, MakeComplex
Images need to be standardized, normalized, and casted to complex data type. Modifiers are tools made for this purpose which apply these operations on import.
p1 = ImageToArray(bit_depth=16, channel='gray', crop_window=None, dtype='float32')
p2 = Normalize(background=np.ones((512, 512)))
p3 = MakeComplex(set_as='amplitude', phase=0)
obj = import_image("images/squares.png", preprocessor=[p1, p2, p3])
- Propagate
Solvers contain propagation algorithms and can be called by solver.solve. In particular, angular Spectrum algorithm convolves the input field with a free-space propagtor function which depends on wavelength λ (or wavenumber k=2π/λ) and distance z.
from numpy import pi, abs, angle
from fringe.solvers.AngularSpectrum import AngularSpectrumSolver as AsSolver
solver = AsSolver(shape=obj.shape, dr=1, is_batched=False, padding="same", pad_fill_value=0, backend="Numpy")
rec = solver.solve(hologram, k=2*pi/500e-3, z=-1000)
amp, phase = abs(rec), angle(rec)
ax = pyplot.sublots(2, 1, 1)
ax[0].imshow(abs(obj))
ax[1].imshow(amp)
pyplot.show()
Example notebooks provide further details with 1D and 2D diffraction, GPU acceleration, batch processing, and phase recovery.
License
Fringe is released under the MIT license. See LICENSE for details.
======= History
0.0.1 (2021-02-19)
- First release on PyPI.
0.0.2 (2021-02-25)
- Some bugs and errors fixed
0.0.3 (2021-02-25)
- Minor bugs fixed
0.0.5 (2023-03-15)
- AngularSpectrum.py Propagator issue for small pixels is resolved Near field propagation is now supported
- io.py skimage warnings io is fixed.
0.0.6 (2023-03-16)
- AngularSpectrum.py Bug fixed
0.0.7 (2023-03-16)
- AngularSpectrum.py Positive real exponential bug fixed
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
File details
Details for the file fringe-1.0.7-py2.py3-none-any.whl
.
File metadata
- Download URL: fringe-1.0.7-py2.py3-none-any.whl
- Upload date:
- Size: 18.2 kB
- Tags: Python 2, Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.2 CPython/3.8.10
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | aaba317dbe231f7c56f38ec43d99841c73a88c526f5cbef2753c4d36dbc69216 |
|
MD5 | ca2bab98e05126b05293c0750bc08790 |
|
BLAKE2b-256 | 59ca0c4c2d2e0b6a841619abe215ff76351a4956efbb55df99e62aee572b8600 |