def test_PSNR_tensor(self):
from utility.psnr import psnr
value = psnr(t_ones() * 10, t_zeros(), max_value=255.)
self.assertTrue(h.close_equal(value, 28.1308036086791))
def test_mse_numpy(self):
from utility.psnr import mse
value = mse(n_ones() * 2, n_zeros())
self.assertTrue(h.close_equal(value, 4.))
def test_mse_tensor(self):
from utility.psnr import mse
value = mse(t_ones() * 0.25, t_zeros())
self.assertTrue(h.close_equal(value, 0.0625))
def test_shift_thirds2(self):
output = ds.batch_shift_downscale(h.test_img1(), h.shift_thirds())
self.assertTrue(h.close_equal(output,
h.downscaled_img1_shift_thirds()))