Nathan Shearer nathanshearer

nathanshearer / waifu2x.py

Created May 19, 2019 09:30 — forked from Tydus/waifu2x.py

waifu2x in 15 lines of Python by @marcan42

	import json, sys, numpy as np
	from scipy import misc, signal
	from PIL import Image
	infile, outfile, modelpath = sys.argv[1:]
	model = json.load(open(modelpath))
	im = Image.open(infile).convert("YCbCr")
	im = misc.fromimage(im.resize((2im.size[0], 2im.size[1]), resample=Image.NEAREST)).astype("float32")
	planes = [np.pad(im[:,:,0], len(model), "edge") / 255.0]
	for step in model:
	o_planes = [sum([signal.convolve2d(ip, np.float32(kernel), "valid")