|
|
|
"""Contains the visualizer to visualize images as a GIF.""" |
|
|
|
from PIL import Image |
|
|
|
from ..image_utils import parse_image_size |
|
from ..image_utils import load_image |
|
from ..image_utils import resize_image |
|
from ..image_utils import list_images_from_dir |
|
|
|
__all__ = ['GifVisualizer'] |
|
|
|
|
|
class GifVisualizer(object): |
|
"""Defines the visualizer that visualizes an image collection as GIF.""" |
|
|
|
def __init__(self, image_size=None, duration=100, loop=0): |
|
"""Initializes the GIF visualizer. |
|
|
|
Args: |
|
image_size: Size for image visualization. (default: None) |
|
duration: Duration between two frames, in milliseconds. |
|
(default: 100) |
|
loop: How many times to loop the GIF. `0` means infinite. |
|
(default: 0) |
|
""" |
|
self.set_image_size(image_size) |
|
self.set_duration(duration) |
|
self.set_loop(loop) |
|
|
|
def set_image_size(self, image_size=None): |
|
"""Sets the image size of the GIF.""" |
|
height, width = parse_image_size(image_size) |
|
self.image_height = height |
|
self.image_width = width |
|
|
|
def set_duration(self, duration=100): |
|
"""Sets the GIF duration.""" |
|
self.duration = duration |
|
|
|
def set_loop(self, loop=0): |
|
"""Sets how many times the GIF will be looped. `0` means infinite.""" |
|
self.loop = loop |
|
|
|
def visualize_collection(self, images, save_path): |
|
"""Visualizes a collection of images one by one.""" |
|
height, width = images[0].shape[0:2] |
|
height = self.image_height or height |
|
width = self.image_width or width |
|
pil_images = [] |
|
for image in images: |
|
if image.shape[0:2] != (height, width): |
|
image = resize_image(image, (width, height)) |
|
pil_images.append(Image.fromarray(image)) |
|
pil_images[0].save(save_path, format='GIF', save_all=True, |
|
append_images=pil_images[1:], |
|
duration=self.duration, |
|
loop=self.loop) |
|
|
|
def visualize_list(self, image_list, save_path): |
|
"""Visualizes a list of image files.""" |
|
height, width = load_image(image_list[0]).shape[0:2] |
|
height = self.image_height or height |
|
width = self.image_width or width |
|
pil_images = [] |
|
for filename in image_list: |
|
image = load_image(filename) |
|
if image.shape[0:2] != (height, width): |
|
image = resize_image(image, (width, height)) |
|
pil_images.append(Image.fromarray(image)) |
|
pil_images[0].save(save_path, format='GIF', save_all=True, |
|
append_images=pil_images[1:], |
|
duration=self.duration, |
|
loop=self.loop) |
|
|
|
def visualize_directory(self, directory, save_path): |
|
"""Visualizes all images under a directory.""" |
|
image_list = list_images_from_dir(directory) |
|
self.visualize_list(image_list, save_path) |
|
|
