High bit depth

data_loader.py

@@ -226,7 +226,7 @@ class FileDataset(Dataset):
         if "labels" in sample:
             # return UDP as 4chn XYZV float tensor
             if sample["labels"].dtype != np.float32:
-                sample["labels"] = sample["labels"].astype(np.float32) / 255
+                sample["labels"] = sample["labels"].astype(np.float32) / np.iinfo(sample["labels"].dtype).max
             sample["labels"] = torch.from_numpy(
                 sample["labels"].transpose((2, 0, 1)))
             assert (sample["labels"].dtype == torch.float32)