Function gif_to_nparray

plib/render.py:523–558

Load gif to numpy array. Args: filename: filename of the gif crop_ratio: ratio of part to be cropped crop_dir: direction to be cropped, default left Returns: ndarray (n,h,w), n: number of frames

(
        filename: str,
        crop_ratio: float = 0,
        crop_dir: str = 'left',
)

def gif_to_nparray(
        filename: str,
        crop_ratio: float = 0,
        crop_dir: str = 'left',
) -> np.ndarray:
    """
    Load gif to numpy array.

    Args:
        filename:  filename of the gif
        crop_ratio: ratio of part to be cropped
        crop_dir: direction to be cropped, default left
    Returns:
        ndarray (n,h,w), n: number of frames
    """
    imglist = []
    if not os.path.exists(filename):
        return None

    imageObject = Image.open(filename)
    for frame in range(0, imageObject.n_frames):
        imageObject.seek(frame)
        tmp = imageObject.convert()  # Make without palette
        tmp = np.asarray(tmp) / 255.
        # print(tmp.shape)
        if crop_ratio != 0:
            if crop_dir == 'left':
                tmp = tmp[:, int(tmp.shape[1] * crop_ratio):]
            elif crop_dir == 'right':
                tmp = tmp[:, :int(tmp.shape[1] * crop_ratio)]
            else:
                raise NotImplementedError

        imglist.append(np.expand_dims(tmp, axis=0))
    gifarray = np.concatenate(imglist, axis=0)  # (n,h,w)
    return gifarray


def add_title_to_image(