Upload 20 files

.gitattributes

@@ -33,3 +33,10 @@ 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
+examples/3.jpg filter=lfs diff=lfs merge=lfs -text
+lib/libDocSdk.so filter=lfs diff=lfs merge=lfs -text
+lib/libidlivesdk.so filter=lfs diff=lfs merge=lfs -text
+lib/libopenvino_intel_cpu_plugin.so filter=lfs diff=lfs merge=lfs -text
+lib/libopenvino_onnx_frontend.so.2300 filter=lfs diff=lfs merge=lfs -text
+lib/libopenvino.so.2300 filter=lfs diff=lfs merge=lfs -text
+lib/libopenvino.so.2330 filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,30 @@
+FROM ubuntu:22.04
+RUN ln -snf /usr/share/zoneinfo/$CONTAINER_TIMEZONE /etc/localtime && echo $CONTAINER_TIMEZONE > /etc/timezone
+RUN apt-get update && \
+    apt-get install -y binutils python3 python3-pip python3-opencv && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+COPY --chown=user ./idlivesdk.py .
+COPY --chown=user ./app.py .
+COPY --chown=user ./demo.py .
+COPY --chown=user ./requirements.txt .
+COPY --chown=user ./run.sh .
+COPY --chown=user ./examples ./examples
+COPY --chown=user ./model ./model
+COPY --chown=user ./lib /usr/lib
+COPY --chown=user ./lib/libidlivesdk.so ./lib/libidlivesdk.so
+
+ADD https://huggingface.co/datasets/FaceOnLive/IDL-Models/resolve/main/model.tar.gz .
+RUN tar -xvzf ./model.tar.gz -C . && rm ./model.tar.gz
+
+RUN chmod -R 777 ./model/
+
+RUN pip3 install -r requirements.txt
+RUN chmod a+x run.sh
+CMD ["./run.sh"]
+EXPOSE 7860

app.py

@@ -0,0 +1,132 @@
+import sys
+sys.path.append('.')
+
+import os
+import numpy as np
+import base64
+import json
+import io
+
+from PIL import Image, ExifTags
+from flask import Flask, request, jsonify
+from idlivesdk import getHWID
+from idlivesdk import setLicenseKey
+from idlivesdk import initSDK
+from idlivesdk import processImage
+
+licenseKeyPath = "license.txt"
+license = os.environ.get("LICENSE_KEY")
+
+if license is None:
+    try:
+        with open(licenseKeyPath, 'r') as file:
+            license = file.read().strip()
+    except IOError as exc:
+        print("failed to open license.txt: ", exc.errno)
+    print("License Key: ", license)
+
+hwid = getHWID()
+print("HWID: ", hwid.decode('utf-8'))
+
+ret = setLicenseKey(license.encode('utf-8'))
+print("Set License: ", ret)
+
+ret = initSDK("model".encode('utf-8'))
+print("Init: ", ret)
+
+app = Flask(__name__) 
+
+def apply_exif_rotation(image):
+    try:
+        exif = image._getexif()
+        if exif is not None:
+            for orientation in ExifTags.TAGS.keys():
+                if ExifTags.TAGS[orientation] == 'Orientation':
+                    break
+
+            # Get the orientation value
+            orientation = exif.get(orientation, None)
+
+            # Apply the appropriate rotation based on the orientation
+            if orientation == 3:
+                image = image.rotate(180, expand=True)
+            elif orientation == 6:
+                image = image.rotate(270, expand=True)
+            elif orientation == 8:
+                image = image.rotate(90, expand=True)
+
+    except AttributeError:
+        print("No EXIF data found")
+
+    return image
+
+
+@app.route('/process_image', methods=['POST'])
+def process_image():
+    file = request.files['image']
+
+    try:
+        image = apply_exif_rotation(Image.open(file)).convert('RGB')        
+    except:
+        result = "Failed to open file"
+        response = jsonify({"resultCode": "Error", "result": result})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    image_np = np.asarray(image)
+    result = processImage(image_np, image_np.shape[1], image_np.shape[0])
+
+    if result is None:
+        result = "Failed to process image"
+        response = jsonify({"resultCode": "Error", "result": result})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+    else:
+        result_dict = json.loads(result.decode('utf-8'))
+        response = jsonify({"resultCode": "Ok", "result": result_dict})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+@app.route('/process_image_base64', methods=['POST'])
+def process_image_base64():
+    try:
+        content = request.get_json()
+        base64_image = content['base64']
+
+        image_data = base64.b64decode(base64_image)
+        image = apply_exif_rotation(Image.open(io.BytesIO(image_data))).convert("RGB")
+    except:
+        result = "Failed to parse base64"
+        response = jsonify({"resultCode": "Error", "result": result})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+    image_np = np.asarray(image)
+    result = processImage(image_np, image_np.shape[1], image_np.shape[0])
+
+    if result is None:
+        result = "Failed to process image"
+        response = jsonify({"resultCode": "Error", "result": result})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+    else:
+        result_dict = json.loads(result.decode('utf-8'))
+        response = jsonify({"resultCode": "Ok", "result": result_dict})
+
+        response.status_code = 200
+        response.headers["Content-Type"] = "application/json; charset=utf-8"
+        return response
+
+if __name__ == '__main__':
+    port = int(os.environ.get("PORT", 9000))
+    app.run(host='0.0.0.0', port=port)

demo.py

@@ -0,0 +1,71 @@
+import os
+import gradio as gr
+import requests
+import json
+import io
+import base64
+import cv2
+import numpy as np
+from gradio.components import Image
+
+screenReplayThreshold = 0.5
+portraitReplaceThreshold = 0.5
+printedCopyThreshold = 0.5
+
+def proc_output(result):
+    if result.ok:
+        json_result = result.json()
+        if json_result.get("resultCode") == "Error":
+            return {"status": "error", "result": "failed to process image"}
+
+        process_results = json_result.get("result")
+        status = process_results.get("status")
+        if status == "Ok":
+            screenReply = process_results.get("screenReply")
+            portraitReplace = process_results.get("portraitReplace")
+            printedCopy = process_results.get("printedCopy")
+            detResult = "genuine"
+
+            # Check for "Spoof" condition
+            if screenReply < screenReplayThreshold or portraitReplace < portraitReplaceThreshold or printedCopy < printedCopyThreshold:
+                detResult = "spoof"
+
+            # Update json_result with the modified process_results
+            return {"status": "ok", "data": {"result": detResult, "screenreplay_integrity_score": screenReply, "portraitreplace_integrity_score": portraitReplace, "printedcutout_integrity_score": printedCopy}}
+            
+        return {"status": "error", "result": "document not found!"}
+    else:
+        return {"status": "error", "result": result.text}    
+
+def id_liveness(path):
+    # Convert PIL image to bytes to send in POST request
+    img_bytes = io.BytesIO()
+    path.save(img_bytes, format="JPEG")
+    img_bytes.seek(0)    
+    
+    url = "http://127.0.0.1:9000/process_image"
+    files = {'image': img_bytes}
+    result = requests.post(url=url, files=files)
+    return proc_output(result)
+
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+    # ID Document Liveness Detection
+    Contact us at https://faceonlive.com for issues and support.<br/><br/>
+    ** For security and privacy, kindly refrain from uploading real ID card or credit card information on this platform.
+    """
+    )
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type='pil')           
+            gr.Examples(['examples/1.jpg', 'examples/2.jpg', 'examples/3.jpg'], 
+                        inputs=image_input)
+            process_button = gr.Button("ID Liveness Detection")
+            
+        with gr.Column():
+            json_output = gr.JSON()
+    
+    process_button.click(id_liveness, inputs=image_input, outputs=[json_output])
+
+demo.launch(server_name="0.0.0.0")

idlivesdk.py

@@ -0,0 +1,24 @@
+import os
+
+from ctypes import *
+from numpy.ctypeslib import ndpointer
+
+libPath = os.path.abspath(os.path.dirname(__file__)) + '/lib/idlivesdk.so'
+libidlivesdk = cdll.LoadLibrary(libPath)
+
+getHWID = libidlivesdk.getHWID
+getHWID.argtypes = []
+getHWID.restype = c_char_p
+
+setLicenseKey = libidlivesdk.setLicenseKey
+setLicenseKey.argtypes = [c_char_p]
+setLicenseKey.restype = c_int32
+
+initSDK = libidlivesdk.initSDK
+initSDK.argtypes = [c_char_p]
+initSDK.restype = c_int32
+
+processImage = libidlivesdk.processImage
+processImage.argtypes = [ndpointer(c_ubyte, flags='C_CONTIGUOUS'), c_int32, c_int32]
+processImage.restype = c_char_p
+

lib/libDocSdk.so

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

lib/libidlivesdk.so

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

lib/libopenvino.so.2300

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

lib/libopenvino.so.2330

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

lib/libopenvino_intel_cpu_plugin.so

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

lib/libopenvino_onnx_frontend.so.2300

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

lib/plugins.xml

@@ -0,0 +1,6 @@
+<ie>
+    <plugins>
+        <plugin name="CPU" location="libopenvino_intel_cpu_plugin.so">
+        </plugin>
+    </plugins>
+</ie>

requirements.txt

@@ -0,0 +1,6 @@
+flask
+flask-cors
+gradio==5.4.0
+datadog_api_client
+opencv-python
+numpy==1.24.4

run.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+cd /app
+exec python3 app.py &
+sleep 10
+exec python3 demo.py