IanFooMZ / SonyBayerFilter

Other
1 stars 0 forks source link

vipdopt: Volumetric Inverse Photonic Design Optimizer

Vipdopt is a package for streamlining the inverse deisgn of nanophotonic devices. This package largely serves as a Python-based wrapper for Lumerical. Vipdopt also provides an interactable GUI for creating optimizations and monitoring their progress.

This code performs an inverse-design optimization based on the adjoint optimization technique ^2 that designs a freeform multilayer optical Bayer filter and focusing lens as described in ^1. The Bayer filter is multiwavelength and focuses and sorts different spectral bands into different prescribed locations on the focal plane. Otherwise known as a color router, this code is set by default to create a 10-layer filter for the visible spectral range, with lateral and vertical dimensions of 2.04 microns. Different options are available in the config to, for example, enable polarization sorting or adjust the spectral bands as necessary.

Tutorials, Examples, and Documentation

See docs/

Requirements

If creating a conda environment, there is not official PySide6 package yet. After activating your environment, you will need to install PySide6 with pip.

Usage

To run an optimization, Vipdopt requires a project_directory containing two files:

More details regarding configuration files are located here.

To run an optimization from a pre-existing project directory, run:

python vipdopt optimize path/to/project/directory

To see more options, run

python vipdopt --help

Outputs

All created files and logs are packaged and output to the selected project folde by default. The internal folder structure is as follows:

The folder opt_info collects information about optical figures of merit, the latest 3-D permittivity values of the designed device, the adjoint gradient, and contains auto-generated plots of the evolution of these values throughout the optimization.
(Please note that values such as transmission are not normalized rigorously for these auto-generated plots, as the inclusion of the proper monitors in Lumerical for rigor would slow down each optimization iteration greatly.)

A log file is also created in the main project folder, named with the specified name using the --log option (defaults to dev.log).

GUI

The GUI can be started by running

python vipdopt gui

This will launch a dashboard that can monitor the progress of a optimization.

A Screenshot of an example optimization loaded in the GUI

From the dashboard you can also open a dialog for editing [^3] the optimization parameters

[^3]: Editing and saving an optimization though the GUI is not yet fully supported. However you can still use this window to view the various parameters.

A .GIF demo using the GUI

While starting and stopping the optimization through the GUI is not completely supported yet, there are some experimental features pertaining to those functions.

Clicking the "Start Optimization" button will create a slurm script that can be submitted to run the optimization. To alter the output file format, edit vipdopt/submit.sh. The parameters of running the job are included in this file according to the SLURM Documentation. More information can be found there.