demo.py

# -*- coding:utf-8 -*-
# Author: RubanSeven

import cv2
import imageio
from augment import distort, stretch, perspective


def create_gif(image_list, gif_name, duration=0.1):
    frames = []
    for image in image_list:
        frames.append(image)
    imageio.mimsave(gif_name, frames, 'GIF', duration=duration)
    return


if __name__ == '__main__':
    im = cv2.imread("imgs/demo.png")
    im = cv2.resize(im, (200, 64))
    cv2.imshow("im_CV", im)
    distort_img_list = list()
    stretch_img_list = list()
    perspective_img_list = list()
    for i in range(12):
        distort_img = distort(im, 4)
        distort_img_list.append(distort_img)
        cv2.imshow("distort_img", distort_img)

        stretch_img = stretch(im, 4)
        cv2.imshow("stretch_img", stretch_img)
        stretch_img_list.append(stretch_img)

        perspective_img = perspective(im)
        cv2.imshow("perspective_img", perspective_img)
        perspective_img_list.append(perspective_img)
        cv2.waitKey(100)

    create_gif(distort_img_list, r'imgs/distort.gif')
    create_gif(stretch_img_list, r'imgs/stretch.gif')
    create_gif(perspective_img_list, r'imgs/perspective.gif')