search

a-lemus96 / fs-nerf

PyTorch implementation for experimenting with frequency regularized Neural Radiance Fields.
1 stars 0 forks source link

Compute SSIM using torch ignite #24

Open a-lemus96 opened 12 months ago

a-lemus96 commented 12 months ago

Compute SSIM using torch ignite library to allow for batch computation of samples. This is intended to avoid CPU-GPU synchronization and serialization when computing these metrics.