安装:
pip install imageio
具体代码
import os
import imageio
def create_gif(image_list, gif_name):
frames = []
for image_name in image_list:
if image_name.endswith('.png'):
print(image_name)
frames.append(imageio.imread(image_name))
# Save them as frames into a gif
imageio.mimsave(gif_name, frames, 'GIF', duration = 0.4) # duration 是帧时长
return
def main():
path=r'/home/xxx/eval/plots/'#存放PNG图片文件夹位置
gif_name = 'curve_gif.gif'#生成gif的名称
files = os.listdir(path)
#files.sort()
files.sort(key = lambda x:int(x[4:-6]), reverse=True)
files = [i for i in files if '0.png' in i]
print('included files: ', files)
image_list=[os.path.join(path,img) for img in files]
create_gif(image_list, gif_name)
if __name__ == "__main__":
main()