Face Recognition API 만들기

JSON으로 요청하면 결과값을 Return 해준다.

import cv2
import json
import face_recognition
from flask import Flask, request, jsonify

def face_recog(img_path):
    img = cv2.imread(img_path, cv2.IMREAD_UNCHANGED)

    face_locations = face_recognition.face_locations(img, model="cnn")

    if len(face_locations) > 0:
        top, right, bottom, left = face_locations[0]  
        face_img = img[top:bottom,left:right]
        feature = face_recognition.face_encodings(img, face_locations)[0]
        return feature.tolist()
    else:
        return 0

app = Flask(__name__)
@app.route("/api/encoding", methods=['POST'])
def test():
    #print(request.json)

    custid = request.json['custid']
    path = request.json['img_path']
    feature = face_recog(path)
  
    return jsonify({'custid':custid, 'feature':feature})
    
if __name__ == "__main__":
    app.run(host="0.0.0.0", port="8001")