search

fnzhan / EMLight

[AAAI 2021] EMLight: Lighting Estimation via Spherical Distribution Approximation, [TIP] GMLight: Lighting Estimation via Geometric Distribution Approximation
168 stars 35 forks source link

Running inference only? #27

Closed DaLizardWizard closed 1 year ago

DaLizardWizard commented 1 year ago

I have been trying to get the code to work on generate a rough HDRI from an image input but have had no luck so far, I have downloaded the pretrained model but I am not entirely sure where to start. Can anyone help me?

fnzhan commented 1 year ago

Please check EMLight/RegressionNetwork/test.py.

But if you are using images different from Laval Indoor dataset, you may need to train the model with your own data thanks to the domain gap.