├── .gitignore ├── README.md ├── data ├── __init__.py ├── dataloader.py └── dataset.py ├── datasets ├── ilt_finetune │ ├── test │ │ └── layout_test.npz │ └── train │ │ └── layout_train.npz ├── ilt_pretrain │ ├── layout │ │ └── layouts.npz │ └── mask │ │ └── masks.npz └── litho_simul │ ├── test │ ├── aerial_image │ │ └── aerial_image_test.npz │ └── mask │ │ └── mask_test.npz │ └── train │ ├── aerial_image │ └── aerial_image_train.npz │ └── mask │ └── mask_train.npz ├── environment.yml ├── imgs ├── actual_layout.png ├── aerial_image.png ├── ref_flow.png ├── synthesized_mask.png └── target_layout.png ├── models ├── __init__.py ├── base_model.py ├── ilt_model.py ├── lithosimul_model.py └── networks.py ├── options ├── __init__.py ├── base_options.py ├── test_options.py └── train_options.py ├── requirements.txt ├── scripts ├── test_ilt.sh ├── test_lithosimul.sh ├── train_ilt_finetune.sh ├── train_ilt_pretrain.sh └── train_lithosimul.sh ├── test_ilt.py ├── test_lithosimul.py ├── train.py └── util ├── MaskArray2Gds.py ├── RandomClip ├── RandomClip.py ├── RandomClipParams.json └── run.sh ├── util.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/data/dataloader.py -------------------------------------------------------------------------------- /data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/data/dataset.py -------------------------------------------------------------------------------- /datasets/ilt_finetune/test/layout_test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/ilt_finetune/test/layout_test.npz -------------------------------------------------------------------------------- /datasets/ilt_finetune/train/layout_train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/ilt_finetune/train/layout_train.npz -------------------------------------------------------------------------------- /datasets/ilt_pretrain/layout/layouts.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/ilt_pretrain/layout/layouts.npz -------------------------------------------------------------------------------- /datasets/ilt_pretrain/mask/masks.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/ilt_pretrain/mask/masks.npz -------------------------------------------------------------------------------- /datasets/litho_simul/test/aerial_image/aerial_image_test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/litho_simul/test/aerial_image/aerial_image_test.npz -------------------------------------------------------------------------------- /datasets/litho_simul/test/mask/mask_test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/litho_simul/test/mask/mask_test.npz -------------------------------------------------------------------------------- /datasets/litho_simul/train/aerial_image/aerial_image_train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/litho_simul/train/aerial_image/aerial_image_train.npz -------------------------------------------------------------------------------- /datasets/litho_simul/train/mask/mask_train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/datasets/litho_simul/train/mask/mask_train.npz -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/environment.yml -------------------------------------------------------------------------------- /imgs/actual_layout.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/imgs/actual_layout.png -------------------------------------------------------------------------------- /imgs/aerial_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/imgs/aerial_image.png -------------------------------------------------------------------------------- /imgs/ref_flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/imgs/ref_flow.png -------------------------------------------------------------------------------- /imgs/synthesized_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/imgs/synthesized_mask.png -------------------------------------------------------------------------------- /imgs/target_layout.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/imgs/target_layout.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/ilt_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/models/ilt_model.py -------------------------------------------------------------------------------- /models/lithosimul_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/models/lithosimul_model.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/models/networks.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/options/__init__.py -------------------------------------------------------------------------------- /options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/options/base_options.py -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/options/train_options.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/test_ilt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/scripts/test_ilt.sh -------------------------------------------------------------------------------- /scripts/test_lithosimul.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/scripts/test_lithosimul.sh -------------------------------------------------------------------------------- /scripts/train_ilt_finetune.sh: -------------------------------------------------------------------------------- 1 | python train.py --phase finetune 2 | -------------------------------------------------------------------------------- /scripts/train_ilt_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/scripts/train_ilt_pretrain.sh -------------------------------------------------------------------------------- /scripts/train_lithosimul.sh: -------------------------------------------------------------------------------- 1 | python train.py --phase litho_simul 2 | -------------------------------------------------------------------------------- /test_ilt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/test_ilt.py -------------------------------------------------------------------------------- /test_lithosimul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/test_lithosimul.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/train.py -------------------------------------------------------------------------------- /util/MaskArray2Gds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/util/MaskArray2Gds.py -------------------------------------------------------------------------------- /util/RandomClip/RandomClip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/util/RandomClip/RandomClip.py -------------------------------------------------------------------------------- /util/RandomClip/RandomClipParams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/util/RandomClip/RandomClipParams.json -------------------------------------------------------------------------------- /util/RandomClip/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/util/RandomClip/run.sh -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/util/util.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinming-wei/LithoNet/HEAD/util/visualizer.py --------------------------------------------------------------------------------