Spaces:

RoCobo
/

WiggleGAN

Runtime error

Rodrigo_Cobo

new way of creating mp4

604f38b over 1 year ago

No virus

3.11 kB

	import os
	from PIL import Image
	import argparse
	import imageio
	import numpy as np
	import cv2
	import mediapy as media

	parser = argparse.ArgumentParser(description='change to useful name')
	parser.add_argument('--dim', default=256, type=int, help='dimention image')
	args = parser.parse_args()

	path = "."
	ROOT_DIR = os.path.dirname(__file__)
	dirs = os.listdir(ROOT_DIR)

	dim = args.dim


	def createVideo(array_pil, space_time = 5, name = "out_"):

	ar_im = []
	for j in range(3):
	for image in array_pil:
	for i in range (0,space_time):
	ar_im.append(np.array(image))

	media.write_video(name +".mp4" , ar_im, fps=30, qp=18)


	def gif_order (data, center=True):
	gif = []
	base = 1

	#primera mitad
	i = int((len(data)-2)/2)
	while(i > base ):
	gif.append(data[i])
	#print(i)
	i -= 1


	#el del medio izq
	gif.append(data[int((len(data)-2)/2) + 1])
	#print(int((len(data)-2)/2) + 1)

	#el inicial
	if center:
	gif.append(data[0])
	#print(0)

	# el del medio der
	gif.append(data[int((len(data) - 2) / 2) + 2])
	#print(int((len(data) - 2) / 2) +2)
	#segunda mitad
	i = int((len(data)-2)/2) + 3
	while (i < len(data)):
	gif.append(data[i])
	#print(i)
	i += 1
	#print("---------")

	invertedgif = gif[::-1]
	invertedgif = invertedgif[1:]

	gif = gif[1:] + invertedgif
	#print(gif)
	#for image in gif:
	# image.show()
	#gsdfgsfgf
	return gif

	print(" [*] Init gif generations!")
	# This would print all the files and directories
	for file in dirs:
	if ".jpg" in file or ".png" in file:
	rowImages = []
	im = Image.open("./" + ROOT_DIR + '/' + file)
	width, height = im.size
	#im = im.convert('P')

	#CROP (left, top, right, bottom)

	pointleft = 3
	pointtop = 3
	i = 0
	while (pointtop < height):
	while (pointleft < width):
	im1 = im.crop((pointleft, pointtop, dim+pointleft, dim+pointtop))
	#h1 = im1.size[0] * 4
	#im1 = im1.resize([h1,h1], Image.Resampling.LANCZOS)
	rowImages.append(im1)
	#im1.show()
	pointleft+= dim+4
	# Ya tengo todas las imagenes podria hacer el gif aca
	rowImages = gif_order(rowImages,center=False)
	createVideo(rowImages, 3, name = "./" + ROOT_DIR + '/' + file[:-4] + "_" + str(i))

	name = "./" + ROOT_DIR + '/' + file[:-4] + "_" + str(i) + '.gif'
	rowImages[0].save(name, save_all=True,format='GIF', append_images=rowImages[1:], optimize=True, duration=100, loop=0)


	#print(rowImages)
	pointtop += dim + 4
	pointleft = 3
	rowImages = []
	i+=1
	#im2 = im.crop((width / 2, 0, width, height))
	# im2.show()

	#im1.save("./2" + file[:-4] + ".png")
	#im2.save("./" + file[:-4] + ".png")


	# Deleted
	#os.remove("data/" + file)
	print(" [*] End gif generations!")