We recommend using python3 and a virtual environment

virtualenv -p python3 .env
source .env/bin/activate
pip install -r requirements.txt

When you're done working on the project, deactivate the virtual environment with deactivate.

A matlab engine for python is needed for EM simulation. Please refer to MathWorks Pages for installation.

Path of RETICOLO should be added in the main.py

You can change the parameters by editing Params.json in results folder.

If you want to train the network, simply run

python main.py 

or

python main.py --output_dir results --wavelength 900 --angle 60

to specify non-default output folder or parameters

All results will store in output_dir/ folder.

-figures/  (figures of generated devices and loss function curve)

-model/    (all weights of the generator)

-outputs/  (500 generated devices in `.mat` format)

-history.mat

-train.log

If you use this code for your research, please cite:

Simulator-based training of generative models for the inverse design of metasurfaces.
Jiaqi Jiang, Jonathan A. Fan

Global Optimization of Dielectric Metasurfaces Using a Physics-Driven Neural Network.
Jiaqi Jiang, Jonathan A. Fan