UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2025-04-05 05:29:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge ce9e8925e3 into 54c035f783

Browse Source
This commit is contained in:
Ziyi Yang 2024-10-30 10:11:32 -04:00 committed by GitHub
commit 05a3e21fb3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
metrics.py
View File

@ -15,7 +15,8 @@ from PIL import Image
import torch
import torchvision.transforms.functional as tf
from utils.loss_utils import ssim
from lpipsPyTorch import lpips
#from lpipsPyTorch import lpips
import lpips
import json
from tqdm import tqdm
from utils.image_utils import psnr
@ -71,7 +72,7 @@ def evaluate(model_paths):
for idx in tqdm(range(len(renders)), desc="Metric evaluation progress"):
ssims.append(ssim(renders[idx], gts[idx]))
psnrs.append(psnr(renders[idx], gts[idx]))
lpipss.append(lpips(renders[idx], gts[idx], net_type='vgg'))
lpipss.append(lpips_fn(renders[idx], gts[idx]))
print(" SSIM : {:>12.7f}".format(torch.tensor(ssims).mean(), ".5"))
print(" PSNR : {:>12.7f}".format(torch.tensor(psnrs).mean(), ".5"))
@ -95,6 +96,7 @@ def evaluate(model_paths):
if __name__ == "__main__":
device = torch.device("cuda:0")
torch.cuda.set_device(device)
lpips_fn = lpips.LPIPS(net='vgg').to(device)
# Set up command line argument parser
parser = ArgumentParser(description="Training script parameters")