diff --git a/test/test_functional_tensor.py b/test/test_functional_tensor.py index d23930e7313..c83eefb990c 100644 --- a/test/test_functional_tensor.py +++ b/test/test_functional_tensor.py @@ -1,6 +1,7 @@ import unittest import random import colorsys +import math from PIL import Image from PIL.Image import NEAREST, BILINEAR, BICUBIC @@ -114,7 +115,13 @@ def test_rgb2hsv(self): colorsys_img = torch.tensor(hsv, dtype=torch.float32) - max_diff = (colorsys_img - ft_hsv_img).abs().max() + ft_hsv_img_h, ft_hsv_img_sv = torch.split(ft_hsv_img, [1, 2], dim=1) + colorsys_img_h, colorsys_img_sv = torch.split(colorsys_img, [1, 2], dim=1) + + max_diff_h = ((colorsys_img_h * 2 * math.pi).sin() - (ft_hsv_img_h * 2 * math.pi).sin()).abs().max() + max_diff_sv = (colorsys_img_sv - ft_hsv_img_sv).abs().max() + max_diff = max(max_diff_h, max_diff_sv) + self.assertLess(max_diff, 1e-5) def test_adjustments(self):