def tensor_to_image()

in src/src/predictor.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def tensor_to_image(tensor):
    """
        Transform tensor to image.
    """
    tensor = tensor * 255
    tensor = np.array(tensor, dtype=np.uint8)
    if np.ndim(tensor) > 3:
        assert tensor.shape[0] == 1
        tensor = tensor[0]
    return Image.fromarray(tensor)