import cv2
import imageio
import numpy as np
import os
def make_gif():
buff = []
for filename in os.listdir("/home/dfy/Speech_Enhancement/angle_images"):
img3 = cv2.imread("/home/dfy/Speech_Enhancement/angle_images/"+filename)
img3=cv2.cvtColor(img3,cv2.COLOR_BGR2RGB)
buff.append(img3)
imageio.mimsave("{}.gif".format("2"), np.array(buff) , duration=0.2)
if __name__ == '__main__':
make_gif()
import os
import numpy as np
from PIL import Image, ImageFont
from PIL import ImageDraw
import pandas
def 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) * 255
pivot_h, pivot_w = (max_h-h)//2, (max_w-w)//2
blank[pivot_h:pivot_h+h,pivot_w:pivot_w+w,:c] = image
images[idx] = blank
try:
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:
pass
imageio.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="")