Class ToTensor

x2ct_nerf/preprocessing/X2CT_transform_3d.py:181–191

To Torch Tensor img: 3D, (z, y, x) or (D, H, W) Returns: img: 3d array, (z,y,x) or (D, H, W)

    return img

class ToTensor(object):
  '''
  To Torch Tensor
  img: 3D, (z, y, x) or (D, H, W)
    Returns:
      img: 3d array, (z,y,x) or (D, H, W)
  '''
  def __call__(self, img):
    img = torch.from_numpy(img.astype(np.float32))

    return img

class Normalization(object):
  '''