Slide 60
Slide 60 text
'MPZE)VCʹΑΔσϓϩΠͷਤ
SFGIUUQTEPDTqPZEIVCDPNDPNNBOETSVOTFSWF
%BUBTFU
NPVOUBCMFGPMEFST
1SFWJPVTUSBJOJOHKPCT
&OEQPJOU
BQQQZ
import os
from flask import Flask, send_file, request
from werkzeug.exceptions import BadRequest
from werkzeug.utils import secure_filename
from evaluate import ffwd_to_img
ALLOWED_EXTENSIONS = set(['jpg', 'jpeg'])
app = Flask(__name__)
@app.route('/', methods=["POST"])
def style_transfer(path):
"""
Take the input image and style transfer it
"""
# check if the post request has the file part
if 'file' not in request.files:
return BadRequest("File not present in request")
file = request.files['file']
if file.filename == '':
return BadRequest("File name is not present in request")
if not allowed_file(file.filename):
return BadRequest("Invalid file type")
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
input_filepath = os.path.join('./images/', filename)
output_filepath = os.path.join('/output/', filename)
file.save(input_filepath)
# Get checkpoint filename from la_muse
checkpoint = request.form.get("checkpoint") or "la_muse.ckpt"
ffwd_to_img(input_filepath, output_filepath, '/input/' + checkpoint, '/gpu:0')
return send_file(output_filepath, mimetype='image/jpg')
def allowed_file(filename):
return '.' in filename and \
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
if __name__ == '__main__':
app.run(host='0.0.0.0')
BQQQZ
NPVOU
qPZESVOFOWLFSBTQZ
EBUBQBUIUPNPEFMNPEFMT
NPEFTFSWF
'MPZEFYQFDUTUIJTpMF
UPDPOUBJOUIFDPEFUP
SVOBXFCTFSWFSBOE
MJTUFOPOQPSU
IUUQTEPDTqPZEIVCDPN
FYBNQMFTTUZMF@USBOTGFS
TFSWFNPEF
https://
www.floydhub.com/
expose/ABCDEFabcdef
SPVUF
ॻ͘