python生成gif

2021/4/22 20:28:40

本文主要是介绍python生成gif,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

import osimport numpy as npfrom PIL import Image, ImageFontfrom PIL import ImageDrawimport pandasdef make_gif(paths, gif_path, max_frame=50, prefix=""):import imageio
    paths=paths+"/"+pandas.DataFrame(os.listdir(paths))paths=paths.values.tolist()pa=[]for pat in paths:if "png" in pat[0]:pa.append(pat[0])skip_frame = len(pa) // max_frame
    pa = pa[::skip_frame+1]images = [imageio.imread(path) for path in pa]max_h, max_w, max_c = np.max(np.array([image.shape for image in images]), 0)for idx, image in enumerate(images):h, w, c = image.shape
        blank = np.ones([max_h, max_w, max_c], dtype=np.uint8) * 255pivot_h, pivot_w = (max_h-h)//2, (max_w-w)//2blank[pivot_h:pivot_h+h,pivot_w:pivot_w+w,:c] = image

        images[idx] = blanktry:imagess = []for image in images:imagess.append(Image.fromarray(image))draws=[]for image in imagess:draws.append(ImageDraw.Draw(image))font = ImageFont.truetype("assets/arial.ttf", 30)for step, draw in enumerate(draws):draw.text((max_h//20, max_h//20),  f"{prefix}step: {format(step, ',d')}", (0, 0, 0), font=font)except IndexError:passimageio.mimsave(gif_path, [np.array(img) for img in images], duration=0.5)if __name__ == '__main__':make_gif("/home/chenyang/PycharmProjects/ENAS-pytorch-master/logs/ptb_2020-04-15_10-00-07/networks","s.gif" , max_frame=50, prefix="")

在这里插入图片描述



这篇关于python生成gif的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程