From 49c6e2c608f118c685cc6d6936854f89a429fd93 Mon Sep 17 00:00:00 2001 From: protsenkovi Date: Mon, 13 May 2024 17:42:51 +0400 Subject: [PATCH] Fixed transfer lut bug --- src/common/lut.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/common/lut.py b/src/common/lut.py index 3c7a4b6..9791197 100644 --- a/src/common/lut.py +++ b/src/common/lut.py @@ -12,7 +12,7 @@ class Domain4DValues(Dataset): def __init__(self, quantization_interval=1): super(Domain4DValues, self).__init__() values1d = torch.arange(0, 256, quantization_interval, dtype=torch.uint8) - values1d = torch.cat([values1d, torch.tensor([255])]) + values1d = torch.cat([values1d, torch.tensor([256])]) self.quantization_interval = quantization_interval self.values = torch.cartesian_prod(*([values1d]*4)).view(-1, 1, 2, 2)