Fixed transfer lut bug

main
protsenkovi 6 months ago
parent a2542c7b1f
commit 49c6e2c608

@ -12,7 +12,7 @@ class Domain4DValues(Dataset):
def __init__(self, quantization_interval=1): def __init__(self, quantization_interval=1):
super(Domain4DValues, self).__init__() super(Domain4DValues, self).__init__()
values1d = torch.arange(0, 256, quantization_interval, dtype=torch.uint8) 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.quantization_interval = quantization_interval
self.values = torch.cartesian_prod(*([values1d]*4)).view(-1, 1, 2, 2) self.values = torch.cartesian_prod(*([values1d]*4)).view(-1, 1, 2, 2)

Loading…
Cancel
Save