def to_raster()

in graphics.py [0:0]

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

def to_raster(x, rescale=False, width=None):
    if len(x.shape) == 3:
        x = x.reshape((x.shape[0], x.shape[1], x.shape[2], 1))
    if x.shape[3] == 1:
        x = np.repeat(x, 3, axis=3)
    if rescale:
        x = (x - x.min()) / (x.max() - x.min()) * 255.
    x = np.clip(x, 0, 255)
    assert len(x.shape) == 4
    assert x.shape[3] == 3
    n_batch = x.shape[0]
    if width is None:
        width = int(np.ceil(np.sqrt(n_batch)))  # result width
    height = int(n_batch / width)  # result height
    tile_height = x.shape[1]
    tile_width = x.shape[2]
    result = np.zeros((int(height * tile_height),
                       int(width * tile_width), 3), dtype='uint8')
    for i in range(height):
        for j in range(width):
            result[i * tile_height:(i + 1) * tile_height, j *
                   tile_width:(j + 1) * tile_width] = x[width*i+j]
    return result