Skip to content

DiffusionLight/DiffusionLight-LoRA-Trainer

Repository files navigation

DiffusionLight-LoRA-Trainer: a code for training LoRA use in DIffusionLight

We provide a code and training set for training LoRA using in DiffusionLight.

Table of contents



Getting started

bash final_train_cont_largeT@900.sh

wait over a night on Single RTX 3090 until it create folder ThisIsTheFinal-lora-hdr-continuous-largeT@900/0_-5/checkpoint-2500 which we provide in main repository

Dataset

We use the Text2Light to generate HDR Environment map using the prompt which also generated by ChatGPT-3.5.

However, you can download a pre-generated dataset at OneDrive (133 GB) which contain 1412 images pair of Low-FOV LDR and HDR Environment map

Installation

Please follow the instructions in our main repository's installation guide

Citation

@inproceedings{Phongthawee2023DiffusionLight,
    author = {Phongthawee, Pakkapon and Chinchuthakun, Worameth and Sinsunthithet, Nontaphat and Raj, Amit and Jampani, Varun and Khungurn, Pramook and Suwajanakorn, Supasorn},
    title = {DiffusionLight: Light Probes for Free by Painting a Chrome Ball},
    booktitle = {ArXiv},
    year = {2023},
}

Visit us 🦉

Vision & Learning Laboratory VISTEC - Vidyasirimedhi Institute of Science and Technology

About

Training code for LoRA using in DiffusionLight

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published