def test():
curr_time = time.time()
image_path = args.path
source = load_image(image_path)
adjusted_image = preprocess(source, IMG_WIDTH, IMG_HEIGTH)
image = denormalize(adjusted_image)
result = generate_image(model, adjusted_image)
image = Image.fromarray(result)
image_path = './images/'
filename = 'result_image.png'
image.save(image_path + filename)
def get_result():
if request.method == "POST":
width, height = 28, 28
try:
source = Image.open(request.files["source"])
adjusted_img = preprocess(source, width, height)
res = predict_number(model, adjusted_img, width, height)
except Exception as e:
print("error : %s" % e)
return Response("fail", status=400)
return str(res)
def get_result():
if request.method == "POST":
width, height = 512, 512
try:
source = request.files['source'].read()
adjusted_image = preprocess(source, width, height)
image = denormalize(adjusted_image)
result = generate_image(model, adjusted_image)
image = Image.fromarray(result)
image_path = './images/'
filename = 'result_image.png'
image.save(image_path + filename)
return send_from_directory(image_path,
filename,
as_attachment=True)
except Exception as e:
print("error : %s" % e)
return Response("fail", status=400)