gitextract_g3mq9mkn/

├── .gitignore
├── LICENSE.txt
├── README.md
├── data/
│   └── .gitignore
├── datasets/
│   ├── UPS_Custom_Dataset.py
│   ├── UPS_DiLiGenT_main.py
│   ├── UPS_Synth_Dataset.py
│   ├── __init__.py
│   ├── custom_data_loader.py
│   ├── pms_transforms.py
│   └── util.py
├── eval/
│   ├── run_stage1.py
│   └── run_stage2.py
├── main_stage1.py
├── main_stage2.py
├── models/
│   ├── LCNet.py
│   ├── NENet.py
│   ├── __init__.py
│   ├── custom_model.py
│   ├── model_utils.py
│   └── solver_utils.py
├── options/
│   ├── __init__.py
│   ├── base_opts.py
│   ├── run_model_opts.py
│   ├── stage1_opts.py
│   └── stage2_opts.py
├── scripts/
│   ├── DiLiGenT_objects.txt
│   ├── cropDiLiGenTData.py
│   ├── download_pretrained_models.sh
│   ├── download_synthetic_datasets.sh
│   └── prepare_diligent_dataset.sh
├── test_stage1.py
├── test_stage2.py
├── train_stage1.py
├── train_stage2.py
└── utils/
    ├── __init__.py
    ├── eval_utils.py
    ├── logger.py
    ├── recorders.py
    ├── time_utils.py
    └── utils.py