From dfa4cb00ce4afe568228674c8369dc39d8156d93 Mon Sep 17 00:00:00 2001
From: Srijeet Roy <srijeet.11@gmail.com>
Date: Wed, 2 Aug 2023 18:31:08 +0200
Subject: [PATCH] fix SSIM error
---
evaluation/helpers/metrics.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/evaluation/helpers/metrics.py b/evaluation/helpers/metrics.py
index 93b11a4..5fcb4be 100644
--- a/evaluation/helpers/metrics.py
+++ b/evaluation/helpers/metrics.py
@@ -14,8 +14,8 @@ from evaluation.helpers.score_infinity import calculate_FID_infinity_path, calcu
def image_to_tensor(path, sample='all', device='cpu'):
- transform_resize = transforms.Compose([transforms.ToTensor(), transforms.Resize(128), transforms.Lambda(lambda x: (x * 255).type(torch.uint8))])
- transform = transforms.Compose([transforms.ToTensor(), transforms.Lambda(lambda x: (x * 255).type(torch.uint8)) ])
+ transform_resize = transforms.Compose([transforms.ToTensor(), transforms.Resize(128), transforms.Lambda(lambda x: (x * 255))])
+ transform = transforms.Compose([transforms.ToTensor(), transforms.Lambda(lambda x: (x * 255)) ])
filelist = []
for p, subdirs, files in os.walk(path):
for f in files:
--
GitLab