diff --git a/flask_im_ser.py b/flask_im_ser.py new file mode 100644 index 0000000..d7fca49 --- /dev/null +++ b/flask_im_ser.py @@ -0,0 +1,31 @@ +from flask import Flask, request, Response +import jsonpickle +import numpy as np +import cv2 + +# Initialize the Flask application +app = Flask(__name__) + + +# route http posts to this method +@app.route('/api/test', methods=['POST']) +def test(): + r = request + # convert string of image data to uint8 + nparr = np.fromstring(r.data, np.uint8) + # decode image + img = cv2.imdecode(nparr, cv2.IMREAD_COLOR) + + # do some fancy processing here.... + + # build a response dict to send back to client + response = {'message': 'image received. size={}x{}'.format(img.shape[1], img.shape[0]) + } + # encode response using jsonpickle + response_pickled = jsonpickle.encode(response) + + return Response(response=response_pickled, status=200, mimetype="application/json") + + +# start flask app +app.run(host="0.0.0.0", port=5000)