From 00755efaea2e8e8c83db01c4f2f893575238cadd Mon Sep 17 00:00:00 2001
From: Dongheon Lee <dslisleedh@gmail.com>
Date: Wed, 8 Mar 2023 09:49:17 +0900
Subject: [PATCH] Update psnr_ssim.py

Fix input range typo
---
 basicsr/metrics/psnr_ssim.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/basicsr/metrics/psnr_ssim.py b/basicsr/metrics/psnr_ssim.py
index ab03113f8..d0579dc9b 100644
--- a/basicsr/metrics/psnr_ssim.py
+++ b/basicsr/metrics/psnr_ssim.py
@@ -204,8 +204,8 @@ def _ssim_pth(img, img2):
     It is called by func:`calculate_ssim_pt`.
 
     Args:
-        img (Tensor): Images with range [0, 1], shape (n, 3/1, h, w).
-        img2 (Tensor): Images with range [0, 1], shape (n, 3/1, h, w).
+        img (Tensor): Images with range [0, 255], shape (n, 3/1, h, w).
+        img2 (Tensor): Images with range [0, 255], shape (n, 3/1, h, w).
 
     Returns:
         float: SSIM result.