├── .gitignore ├── LICENSE ├── README.md ├── assets ├── examples │ ├── clay_vase.png │ ├── dog_head_desk.png │ ├── read_scene.png │ ├── rum_tank.png │ ├── treasure_box.png │ └── vases_scene.png ├── method_overview.jpg └── teaser-video.gif ├── inference.py ├── mld ├── __init__.py ├── configs │ ├── infer.yaml │ ├── train-phase-1.yaml │ └── train-phase-2.yaml ├── dataset.py ├── model.py ├── pipeline.py ├── train.py └── util.py ├── recon ├── __init__.py ├── callbacks.py ├── dataset.py ├── model.py ├── options │ ├── NLR │ │ ├── __init__.py │ │ ├── train-256.py │ │ └── train-512.py │ └── __init__.py └── trainer.py ├── requirements.txt └── utils ├── __init__.py ├── augs.py ├── proxy.py └── vis.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/README.md -------------------------------------------------------------------------------- /assets/examples/clay_vase.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/examples/clay_vase.png -------------------------------------------------------------------------------- /assets/examples/dog_head_desk.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/examples/dog_head_desk.png -------------------------------------------------------------------------------- /assets/examples/read_scene.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/examples/read_scene.png -------------------------------------------------------------------------------- /assets/examples/rum_tank.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/examples/rum_tank.png -------------------------------------------------------------------------------- /assets/examples/treasure_box.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/examples/treasure_box.png -------------------------------------------------------------------------------- /assets/examples/vases_scene.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/examples/vases_scene.png -------------------------------------------------------------------------------- /assets/method_overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/method_overview.jpg -------------------------------------------------------------------------------- /assets/teaser-video.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/assets/teaser-video.gif -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/inference.py -------------------------------------------------------------------------------- /mld/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mld/configs/infer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/configs/infer.yaml -------------------------------------------------------------------------------- /mld/configs/train-phase-1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/configs/train-phase-1.yaml -------------------------------------------------------------------------------- /mld/configs/train-phase-2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/configs/train-phase-2.yaml -------------------------------------------------------------------------------- /mld/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/dataset.py -------------------------------------------------------------------------------- /mld/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/model.py -------------------------------------------------------------------------------- /mld/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/pipeline.py -------------------------------------------------------------------------------- /mld/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/train.py -------------------------------------------------------------------------------- /mld/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/mld/util.py -------------------------------------------------------------------------------- /recon/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recon/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/recon/callbacks.py -------------------------------------------------------------------------------- /recon/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/recon/dataset.py -------------------------------------------------------------------------------- /recon/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/recon/model.py -------------------------------------------------------------------------------- /recon/options/NLR/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recon/options/NLR/train-256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/recon/options/NLR/train-256.py -------------------------------------------------------------------------------- /recon/options/NLR/train-512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/recon/options/NLR/train-512.py -------------------------------------------------------------------------------- /recon/options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/recon/options/__init__.py -------------------------------------------------------------------------------- /recon/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/recon/trainer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .proxy import * 2 | -------------------------------------------------------------------------------- /utils/augs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/utils/augs.py -------------------------------------------------------------------------------- /utils/proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/utils/proxy.py -------------------------------------------------------------------------------- /utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZexinHe/Neural-LightRig/HEAD/utils/vis.py --------------------------------------------------------------------------------