/NeuralMPS

Primary LanguagePython

NeuralMPS

This is the official implementation of the IJCAI'23 paper "Non-Lambertian Multispectral Photometric Stereo via Spectral Reflectance Decomposition"

Download test data

Please download test data via: https://drive.google.com/drive/folders/1fm0IUYvPOe1GzjG_bxVq-qGSuQB9d_bq?usp=sharing first. Download the folder "test_data" to the main folder "NeuralMPS", unzip it and use the name "test_data".

Dependencies

$ conda install pytorch torchvision cudatoolkit=11.3 -c pytorch -c conda-forge
$ pip install scipy matplotlib imageio scikit-image

Run the test code

Simply run

$ python test.py

This file will use our pretrained model to test the Sphere dataset. Results will be stored in "test_data/results", including the predicted normal map, predicted equivalent light intensity and ground truth equivalent light intensity.

Test your own data

Please follow the data format of "test_data". Or you can refer to "datasets/my_npy_dataloader.py" to write your own dataloader.