search

kakaobrain / nerf-factory

An awesome PyTorch NeRF library
https://kakaobrain.github.io/NeRF-Factory
Apache License 2.0
1.27k stars 107 forks source link

added evaluation metrics: msssim, gmsd, msgmsd, mdsi, haarpsi, vsi, fsim, fid #48

Open dberga opened 7 months ago

dberga commented 7 months ago

Added full-reference metrics (msssim, gmsd, msgmsd, mdsi, haarpsi, vsi, fsim) and feature-based metrics (fid). All functionality has been implemented in src/model/interface.py and exporting its tensorboard test and validation scalars for each existing model in nerf-factory src/model/*/model.py

More metrics could be added from piq library (https://github.com/photosynthesis-team/piq?tab=readme-ov-file#list-of-metrics) but at the moment these were the ones relevant to be tested. Extra metrics from piqa (https://github.com/francois-rozet/piqa?tab=readme-ov-file#available-metrics) could be considered but I retained to use the same library (piq) to avoid any compatibility/dependency conflict.