A napari widget to process Structured Illumination Microscopy data with gpu acceleration
A Napari plugin for the reconstruction of Structured Illumination Microscopy (SIM) with GPU acceleration (with pytorch, if installed). Currently supports:
- conventional SIM data with a generic number of angle and phases (typically, 3 angles and 3 phases are used for resolution improvement in 2D)
- hexagonal SIM data with 7 phases.
The SIM processing widget accepts image stacks organized in 5D (
The reshape widget can be used to easily reshape the data if they are not organized as 5D (angle,phase,z,y,x).
Currently only square images are supported (
For raw image stacks with multiple z-frames each plane is processed as described here: https://doi.org/10.1098/rsta.2020.0162
Support for 3D SIM with enhanced resolution in all directions is not yet available. Multicolor reconstruction is not yet available.
You can install
napari-sim-processor via pip:
pip install napari-sim-processor
To install latest development version :
pip install git+https://github.com/andreabassi78/napari-sim-processor.git
Launch the reshape and sim-processor widgets.
Open your raw image stack (using the napari built-in or your own file openers).
If your image is ordered as a 5D stack (angle, phase, z-frame, y, x) go to point 6.
In the reshape widget, select the actual number of acquired angles, phases, and frames (red arrow) and press
Reshape Stack. Note that the label axis of the viewer will be updated (green arrow).
In the sim-reconstruction widget press the Select image layer button. Note that the number of phases and angles will be updated (blue arrow).
Choose the correct parameters of the SIM acquisition (
M, etc.) and processing parameters (
w: parameter of the Weiner filter.
eta: constant used for calibration. It should be slightly smaller than the carrier frequency (in pupil radius units).
group: for stacks with multiple z-frames, it is the number of frames that are used together for the calibration process.
For details on the other parameters see https://doi.org/10.1098/rsta.2020.0162.
Calibrate the SIM processor, pressing the
Calibratebutton. This will find the carrier frequencies (red circles if the
Show Carriercheckbox is selected), the modulation amplitude and the phase, using cross correlation analysis.
Click on the checkboxes to show the power spectrum of the raw image (
Show power spectrum) or the cross-correlation (
Show Xcorr), to see if the found carrier frequency is correct.
Napari viewer showing the power spectrum of the raw stack. The pupil circle is in blue. A circle corresponding to
eta is shown in green.
Napari viewer showing the cross-correlation of the raw stack. The red circles indicate the found carrier frequencies
- Run the reconstruction of a single plane (
SIM reconstruction) or of a stack (
Stack reconstruction). After execution, a new image_layer will be added to the napari viewer. Click on the
Batch reconstructioncheckbox in order to process an entire stack in one shot. Click on the pytorch checkbox for gpu acceleration.
Napari viewer with widgets showing a pseudo-widefield reconstruction
Napari viewer with widgets showing a SIM reconstruction
The underlying processing classes will use numpy (and FFTW if available) for its calculations. For GPU accelerated processing you need to have either the PyTorch (tested with torch v1.11.0+cu113) or the CuPy (tested with cupy-cuda113 v10.4.0) package installed. Make sure to match the package cuda version to the CUDA library installed on your system otherwise PyTorch will default to CPU and CuPy will not work at all.
Both packages give significant speedup on even relatively modest CUDA GPUs compared to Numpy, and PyTorch running on the CPU only can show improvements relative to numpy and FFTW. Selection of which processing package to use is via a ComboBox in the napari_sim_processor widget. Only available packages are shown.
Other than requiring a CUDA GPU it is advisable to have significant GPU memory available, particularly when processing large datasets. Batch processing is the most memory hungry of the methods, but can process 280x512x512 datasets on a 4GB GPU.
Contributions are very welcome. Tests can be run with tox, please ensure the coverage at least stays the same before you submit a pull request.
Distributed under the terms of the BSD-3 license, "napari-sim-processor" is free and open source software
If you encounter any problems, please file an issue along with a detailed description.
- 22 May 2022
- 04 May 2022
- Stars: 3
- Forks: 1
- Issues + PRs: 2