├── README.md ├── data ├── datasets │ └── get_material_id.py └── models │ ├── GNet_checkp_1.pth.tar │ ├── GNet_checkp_1_rec.pth.tar │ ├── RNet_checkp_1.pth.tar │ └── RNet_checkp_1_rec.pth.tar ├── datasets ├── DiLiGenT_main.py ├── DiLiGenT_main_r2_2.py ├── Dragon.py ├── PS_Synth_Dataset.py ├── __init__.py ├── custom_data_loader.py ├── pms_transforms.py └── util.py ├── eval ├── cal_REL.py └── run_model_Deligent.py ├── list of 100 materials.txt ├── main.py ├── models ├── GR_PSN.py ├── GR_PSN_run.py ├── custom_model.py ├── model_utils.py └── solver_utils.py ├── options ├── base_opts.py ├── run_model_opts.py ├── run_model_opts_Dragon.py ├── run_model_opts_r2.py └── train_opts.py ├── results ├── Dragon-render.mp4 ├── Normal1_1.png ├── Normal1_10.png ├── Normal1_2.png ├── Normal1_3.png ├── Normal1_4.png ├── Normal1_5.png ├── Normal1_6.png ├── Normal1_7.png ├── Normal1_8.png ├── Normal1_9.png ├── errormap1_1-2.78.png ├── errormap1_10-12.39.png ├── errormap1_2-5.31.png ├── errormap1_3-7.03.png ├── errormap1_4-5.63.png ├── errormap1_5-6.72.png ├── errormap1_6-7.11.png ├── errormap1_7-14.08.png ├── errormap1_8-6.54.png └── errormap1_9-6.76.png ├── scripts ├── download_synthetic_datasets.sh └── prepare_diligent_dataset.sh ├── test_utils.py ├── train_utils.py └── utils ├── eval_utils.py ├── logger.py ├── recorders.py ├── time_utils.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/README.md -------------------------------------------------------------------------------- /data/datasets/get_material_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/data/datasets/get_material_id.py -------------------------------------------------------------------------------- /data/models/GNet_checkp_1.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/data/models/GNet_checkp_1.pth.tar -------------------------------------------------------------------------------- /data/models/GNet_checkp_1_rec.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/data/models/GNet_checkp_1_rec.pth.tar -------------------------------------------------------------------------------- /data/models/RNet_checkp_1.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/data/models/RNet_checkp_1.pth.tar -------------------------------------------------------------------------------- /data/models/RNet_checkp_1_rec.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/data/models/RNet_checkp_1_rec.pth.tar -------------------------------------------------------------------------------- /datasets/DiLiGenT_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/datasets/DiLiGenT_main.py -------------------------------------------------------------------------------- /datasets/DiLiGenT_main_r2_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/datasets/DiLiGenT_main_r2_2.py -------------------------------------------------------------------------------- /datasets/Dragon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/datasets/Dragon.py -------------------------------------------------------------------------------- /datasets/PS_Synth_Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/datasets/PS_Synth_Dataset.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/custom_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/datasets/custom_data_loader.py -------------------------------------------------------------------------------- /datasets/pms_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/datasets/pms_transforms.py -------------------------------------------------------------------------------- /datasets/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/datasets/util.py -------------------------------------------------------------------------------- /eval/cal_REL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/eval/cal_REL.py -------------------------------------------------------------------------------- /eval/run_model_Deligent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/eval/run_model_Deligent.py -------------------------------------------------------------------------------- /list of 100 materials.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/list of 100 materials.txt -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/main.py -------------------------------------------------------------------------------- /models/GR_PSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/models/GR_PSN.py -------------------------------------------------------------------------------- /models/GR_PSN_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/models/GR_PSN_run.py -------------------------------------------------------------------------------- /models/custom_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/models/custom_model.py -------------------------------------------------------------------------------- /models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/models/model_utils.py -------------------------------------------------------------------------------- /models/solver_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/models/solver_utils.py -------------------------------------------------------------------------------- /options/base_opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/options/base_opts.py -------------------------------------------------------------------------------- /options/run_model_opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/options/run_model_opts.py -------------------------------------------------------------------------------- /options/run_model_opts_Dragon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/options/run_model_opts_Dragon.py -------------------------------------------------------------------------------- /options/run_model_opts_r2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/options/run_model_opts_r2.py -------------------------------------------------------------------------------- /options/train_opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/options/train_opts.py -------------------------------------------------------------------------------- /results/Dragon-render.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Dragon-render.mp4 -------------------------------------------------------------------------------- /results/Normal1_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_1.png -------------------------------------------------------------------------------- /results/Normal1_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_10.png -------------------------------------------------------------------------------- /results/Normal1_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_2.png -------------------------------------------------------------------------------- /results/Normal1_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_3.png -------------------------------------------------------------------------------- /results/Normal1_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_4.png -------------------------------------------------------------------------------- /results/Normal1_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_5.png -------------------------------------------------------------------------------- /results/Normal1_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_6.png -------------------------------------------------------------------------------- /results/Normal1_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_7.png -------------------------------------------------------------------------------- /results/Normal1_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_8.png -------------------------------------------------------------------------------- /results/Normal1_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/Normal1_9.png -------------------------------------------------------------------------------- /results/errormap1_1-2.78.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_1-2.78.png -------------------------------------------------------------------------------- /results/errormap1_10-12.39.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_10-12.39.png -------------------------------------------------------------------------------- /results/errormap1_2-5.31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_2-5.31.png -------------------------------------------------------------------------------- /results/errormap1_3-7.03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_3-7.03.png -------------------------------------------------------------------------------- /results/errormap1_4-5.63.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_4-5.63.png -------------------------------------------------------------------------------- /results/errormap1_5-6.72.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_5-6.72.png -------------------------------------------------------------------------------- /results/errormap1_6-7.11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_6-7.11.png -------------------------------------------------------------------------------- /results/errormap1_7-14.08.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_7-14.08.png -------------------------------------------------------------------------------- /results/errormap1_8-6.54.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_8-6.54.png -------------------------------------------------------------------------------- /results/errormap1_9-6.76.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/results/errormap1_9-6.76.png -------------------------------------------------------------------------------- /scripts/download_synthetic_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/scripts/download_synthetic_datasets.sh -------------------------------------------------------------------------------- /scripts/prepare_diligent_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/scripts/prepare_diligent_dataset.sh -------------------------------------------------------------------------------- /test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/test_utils.py -------------------------------------------------------------------------------- /train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/train_utils.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/recorders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/utils/recorders.py -------------------------------------------------------------------------------- /utils/time_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/utils/time_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kelvin-Ju/GR-PSN/HEAD/utils/utils.py --------------------------------------------------------------------------------