def __call__(self, image):
    image = np.transpose(image, (2, 0, 1))
    image = np.ascontiguousarray(image, dtype=np.float32)
    tensor = torch.from_numpy(image)
    return tensor.unsqueeze(0)
def __call__(self, image):
    image = np.transpose(image, (2, 0, 1))
    image = np.ascontiguousarray(image, dtype=np.float32)
    tensor = torch.from_numpy(image)
    return tensor.unsqueeze(0)