Upload 10 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+facewrapper/libs/libttvfaceengine6.so filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,16 @@
+FROM ubuntu:20.04
+RUN ln -snf /usr/share/zoneinfo/$CONTAINER_TIMEZONE /etc/localtime && echo $CONTAINER_TIMEZONE > /etc/timezone
+RUN apt-get update -y
+RUN apt-get install -y python3 python3-pip python3-opencv
+RUN apt-get install -y libcurl4-openssl-dev libssl-dev
+RUN mkdir -p /root/FaceOnLive_v6
+RUN mkdir -p /root/FaceOnLive_v6/facewrapper
+WORKDIR /root/FaceOnLive_v6
+COPY ./facewrapper ./facewrapper
+COPY ./facewrapper/libs/libimutils.so /usr/lib
+COPY ./facewrapper/libs/openvino /usr/lib
+COPY ./app.py ./app.py
+COPY ./requirements.txt ./requirements.txt
+RUN pip3 install -r requirements.txt
+CMD [ "python3", "app.py"]
+EXPOSE 8000

app.py

@@ -0,0 +1,198 @@
+import sys
+sys.path.append('.')
+
+from flask import Flask, request, jsonify
+from time import gmtime, strftime
+import os
+import base64
+import json
+import cv2
+import numpy as np
+
+from facewrapper.facewrapper import ttv_version
+from facewrapper.facewrapper import ttv_get_hwid
+from facewrapper.facewrapper import ttv_init
+from facewrapper.facewrapper import ttv_init_offline
+from facewrapper.facewrapper import ttv_extract_feature
+from facewrapper.facewrapper import ttv_compare_feature
+
+app = Flask(__name__) 
+
+app.config['SITE'] = "http://0.0.0.0:8000/"
+app.config['DEBUG'] = False
+
+licenseKey = "IBHTS-SIGHK-VVGCB-MGJYC"
+licensePath = "license.txt"
+modelFolder = os.path.abspath(os.path.dirname(__file__)) + '/facewrapper/dict'
+
+version = ttv_version()
+print("version: ", version.decode('utf-8'))
+
+ret = ttv_init(modelFolder.encode('utf-8'), licenseKey.encode('utf-8'))
+if ret != 0:
+    print(f"online init failed: {ret}");
+
+    hwid = ttv_get_hwid()
+    print("hwid: ", hwid.decode('utf-8'))
+
+    ret = ttv_init_offline(modelFolder.encode('utf-8'), licensePath.encode('utf-8'))
+    if ret != 0:
+        print(f"offline init failed: {ret}")
+        exit(-1)
+    else:
+        print(f"offline init ok")
+
+else:
+    print(f"online init ok")
+
+@app.route('/api/compare_face', methods=['POST'])
+def compare_face():
+    file1 = request.files['image1']
+    image1 = cv2.imdecode(np.fromstring(file1.read(), np.uint8), cv2.IMREAD_COLOR)
+    if image1 is None:
+        result = "image1: is null!"
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    file2 = request.files['image2']
+    image2 = cv2.imdecode(np.fromstring(file2.read(), np.uint8), cv2.IMREAD_COLOR)
+    if image2 is None:
+        result = "image2: is null!"
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+    
+    faceRect1 = np.zeros([4], dtype=np.int32)
+    feature1 = np.zeros([2048], dtype=np.uint8)
+    featureSize1 = np.zeros([1], dtype=np.int32)
+
+    ret = ttv_extract_feature(image1, image1.shape[1], image1.shape[0], faceRect1, feature1, featureSize1)
+    if ret <= 0:
+        if ret == -1:
+            result = "license error!"
+        elif ret == -2:
+            result = "init error!"
+        elif ret == 0:
+            result = "image1: no face detected!"
+
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    faceRect2 = np.zeros([4], dtype=np.int32)
+    feature2 = np.zeros([2048], dtype=np.uint8)
+    featureSize2 = np.zeros([1], dtype=np.int32)
+
+    ret = ttv_extract_feature(image2, image2.shape[1], image2.shape[0], faceRect2, feature2, featureSize2)
+    if ret <= 0:
+        if ret == -1:
+            result = "license error!"
+        elif ret == -2:
+            result = "init error!"
+        elif ret == 0:
+            result = "image2: no face detected!"
+
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    similarity = ttv_compare_feature(feature1, feature2)
+    if similarity > 0.67:
+        result = "same person!"
+    else:
+        result = "different person!"
+  
+    status = "ok"
+    response = jsonify({"status": status, "data": {"result": result, "similarity": float(similarity)}})
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+
+@app.route('/api/compare_face_base64', methods=['POST'])
+def coompare_face_base64():
+    content = request.get_json()
+    imageBase641 = content['image1']
+    image1 = cv2.imdecode(np.frombuffer(base64.b64decode(imageBase641), dtype=np.uint8), cv2.IMREAD_COLOR)
+
+    if image1 is None:
+        result = "image1: is null!"
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    imageBase642 = content['image2']
+    image2 = cv2.imdecode(np.frombuffer(base64.b64decode(imageBase642), dtype=np.uint8), cv2.IMREAD_COLOR)
+
+    if image2 is None:
+        result = "image2: is null!"
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    faceRect1 = np.zeros([4], dtype=np.int32)
+    feature1 = np.zeros([2048], dtype=np.uint8)
+    featureSize1 = np.zeros([1], dtype=np.int32)
+
+    ret = ttv_extract_feature(image1, image1.shape[1], image1.shape[0], faceRect1, feature1, featureSize1)
+    if ret <= 0:
+        if ret == -1:
+            result = "license error!"
+        elif ret == -2:
+            result = "init error!"
+        elif ret == 0:
+            result = "image1: no face detected!"
+
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    faceRect2 = np.zeros([4], dtype=np.int32)
+    feature2 = np.zeros([2048], dtype=np.uint8)
+    featureSize2 = np.zeros([1], dtype=np.int32)
+
+    ret = ttv_extract_feature(image2, image2.shape[1], image2.shape[0], faceRect2, feature2, featureSize2)
+    if ret <= 0:
+        if ret == -1:
+            result = "license error!"
+        elif ret == -2:
+            result = "init error!"
+        elif ret == 0:
+            result = "image2: no face detected!"
+
+        status = "ok"
+        response = jsonify({"status": status, "data": {"result": result}})
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    similarity = ttv_compare_feature(feature1, feature2)
+    if similarity > 0.67:
+        result = "same person!"
+    else:
+        result = "different person!"
+  
+    status = "ok"
+    response = jsonify({"status": status, "data": {"result": result, "similarity": float(similarity)}})
+    response.status_code = 200
+    response.headers["Content-Type"] = "application/json; charset=utf-8"
+    return response
+
+if __name__ == '__main__':
+    port = int(os.environ.get("PORT", 8000))
+    app.run(host='0.0.0.0', port=port)

facewrapper/dict/data1.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:36cf5fcc49345989a86839a53529314ec1fe5d621c377a1952bc7538d55e7f1b
+size 16255630

facewrapper/dict/data2.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f25fb0cd3d70cb84c258e7109620f411c087e0875828d6ab86cc9c4838d49bec
+size 11875339

facewrapper/dict/data3.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06daf36a8545f59ac104415e8b2d38072d06abc027cb346fd4a6c6029fed55b4
+size 90716260

facewrapper/facewrapper.py

@@ -0,0 +1,32 @@
+import ctypes, ctypes.util
+from ctypes import *
+from numpy.ctypeslib import ndpointer
+import sys
+import os
+
+lib_path = os.path.abspath(os.path.dirname(__file__)) + '/libs/libttvfaceengine6.so'
+liveness_engine = cdll.LoadLibrary(lib_path)
+
+ttv_version = liveness_engine.ttv_version
+ttv_version.argtypes = []
+ttv_version.restype = ctypes.c_char_p
+
+ttv_get_hwid = liveness_engine.ttv_get_hwid
+ttv_get_hwid.argtypes = []
+ttv_get_hwid.restype = ctypes.c_char_p
+
+ttv_init = liveness_engine.ttv_init
+ttv_init.argtypes = [ctypes.c_char_p, ctypes.c_char_p]
+ttv_init.restype = ctypes.c_int32
+
+ttv_init_offline = liveness_engine.ttv_init_offline
+ttv_init_offline.argtypes = [ctypes.c_char_p, ctypes.c_char_p]
+ttv_init_offline.restype = ctypes.c_int32
+
+ttv_extract_feature = liveness_engine.ttv_extract_feature
+ttv_extract_feature.argtypes = [ndpointer(ctypes.c_ubyte, flags='C_CONTIGUOUS'), ctypes.c_int32, ctypes.c_int32, ndpointer(ctypes.c_int32, flags='C_CONTIGUOUS'), ndpointer(ctypes.c_ubyte, flags='C_CONTIGUOUS'), ndpointer(ctypes.c_int32, flags='C_CONTIGUOUS')]
+ttv_extract_feature.restype = ctypes.c_int
+
+ttv_compare_feature = liveness_engine.ttv_compare_feature
+ttv_compare_feature.argtypes = [ndpointer(ctypes.c_ubyte, flags='C_CONTIGUOUS'), ndpointer(ctypes.c_ubyte, flags='C_CONTIGUOUS')]
+ttv_compare_feature.restype = ctypes.c_double

facewrapper/libs/libttvfaceengine6.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47263ed3dae141d4640141d2a91c6ad72934771ff7101ff916f26df62c714952
+size 1233040

requirements.txt

@@ -0,0 +1,3 @@
+opencv-python
+flask
+flask-cors