Fixed transfer lut bug

8 months ago · 49c6e2c608
parent a2542c7b1f
commit 49c6e2c608
1 changed files with 1 additions and 1 deletions
--- 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)