Upload 18 files

.gitattributes

@@ -33,3 +33,5 @@ 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
+arial.ttf filter=lfs diff=lfs merge=lfs -text
+lib/libopymrzenv.so filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.8-slim
+WORKDIR /home/opymrz
+COPY ./opymrz.py .
+COPY ./app.py .
+COPY ./demo.py .
+COPY ./license.txt .
+COPY ./requirements.txt .
+COPY ./run.sh .
+COPY ./arial.ttf .
+COPY ./examples ./examples
+COPY ./model ./model
+COPY ./lib/libopyalpr.so ./lib/libopyalpr.so
+COPY ./lib/libopyalprenv.so ./lib/libopyalprenv.so
+COPY ./lib/libtensorflow_framework.so.2.6.0 /usr/lib/libtensorflow_framework.so.2.6.0
+COPY ./lib/libtensorflow.so.2.6.0 /usr/lib/libtensorflow.so.2.6.0
+RUN ln -s /usr/lib/libtensorflow.so.2.6.0 /usr/lib/libtensorflow.so.2 \
+    && ln -s /usr/lib/libtensorflow.so.2.6.0 /usr/lib/libtensorflow.so \
+    && ln -s /usr/lib/libtensorflow_framework.so.2.6.0 /usr/lib/libtensorflow_framework.so.2 \
+    && ln -s /usr/lib/libtensorflow_framework.so.2.6.0 /usr/lib/libtensorflow_framework.so
+RUN pip3 install -r requirements.txt
+RUN chmod a+x run.sh
+CMD ["./run.sh"]
+EXPOSE 9000 7860

app.py

@@ -0,0 +1,131 @@
+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 opymrz import getHWID
+from opymrz import setLicenseKey
+from opymrz import initSDK
+from opymrz import processImage
+
+licenseKeyPath = "license.txt"
+license = ""
+
+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)

arial.ttf

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

demo.py

@@ -0,0 +1,138 @@
+import os
+import gradio as gr
+import requests
+import json
+import io
+from gradio.components import Image
+from PIL import Image as PILImage, ImageDraw, ImageFont  # This import may be needed if you're processing images
+
+from PIL import Image
+
+def resize_image_with_aspect_ratio(image, target_height):
+    # Get original dimensions
+    original_width, original_height = image.size
+    
+    # Calculate the new width to maintain aspect ratio
+    aspect_ratio = original_width / original_height
+    new_width = int(target_height * aspect_ratio)
+    
+    # Resize the image
+    resized_image = image.resize((new_width, target_height), Image.LANCZOS)
+    return resized_image
+
+def draw_text_with_background(draw, text, position, font, text_color="red", bg_color="white", padding=2):
+    # Get the bounding box of the text
+    bbox = draw.textbbox(position, text, font=font)
+    
+    # Calculate the background rectangle coordinates
+    bg_x0, bg_y0, bg_x1, bg_y1 = bbox[0] - padding, bbox[1] - padding, bbox[2] + padding, bbox[3] + padding
+    
+    # Draw the background rectangle
+    draw.rectangle([bg_x0, bg_y0, bg_x1, bg_y1], fill=bg_color)
+    
+    # Draw the text on top of the background
+    draw.text(position, text, fill=text_color, font=font)
+
+def process_image(image):
+    # Convert PIL image to bytes to send in POST request
+    img_bytes = io.BytesIO()
+    image.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)
+
+    # Font setup (default font)
+    try:        
+        font = ImageFont.truetype("arial.ttf", int(image.size[1] / 30))
+    except IOError:
+        font = ImageFont.load_default()
+
+    draw = ImageDraw.Draw(image)
+    if result.ok:
+        json_result = result.json()
+        if json_result.get("resultCode") == "Error":
+            return [image, {"resultCode": "Error", "result": "Failed to process image"}]
+        
+        # plates = json_result.get("result", {}).get("plates", [])
+        # for plate in plates:
+        #     number = plate.get("number")
+        #     left = plate.get("left")
+        #     top = plate.get("top")
+        #     right = plate.get("right")
+        #     bottom = plate.get("bottom")
+        #     countryCode = plate.get('countryCode')
+        #     car = plate.get('car')        
+            
+        #     if car is not None:
+        #         car_left = car.get('left')
+        #         car_top = car.get('top')
+        #         car_right = car.get('right')
+        #         car_bottom = car.get('bottom')
+        #         color = car.get('color')
+        #         madeBy = car.get('madeBy')
+        #         model = car.get('model')
+        #         draw.rectangle([car_left, car_top, car_right, car_bottom], outline="white", width=3)
+        #         draw_text_with_background(draw, color + "," + madeBy + "," + model, (car_left, car_top - 10), font, text_color="blue", bg_color="white")
+                
+            
+        #     # Draw rectangle and plate number
+        #     if left is not None:
+        #         draw.rectangle([left, top, right, bottom], outline="green", width=3)
+        #         draw_text_with_background(draw, number + "," + countryCode, (left, top - 10), font, text_color="green", bg_color="white")
+
+        
+        return [image, json_result]
+    else:
+        return [image, {"resultCode": "Error", "result": result.text}]
+    
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+        <div style="display: flex;align-items: center;">
+            <img alt="Opulentyn Logo" src="https://github.com/user-attachments/assets/5fc78032-bff2-4f7e-a174-7d64b22f506d" width="350"/>
+            <div>
+                <h1>License Plate Recognition (ANPR/ALPR)</h1>
+                <p>We offer <b>on-premises</b> OCR and liveness check solutions available with a <b>perpetual license</b>.</p>
+            </div>
+        </div>
+        ### 🤝 Talk to us
+        <div style="display: flex; align-items: center;">
+            <a href="https://opulentyn.com" target="_blank">
+                <img src="https://img.shields.io/badge/Website-https%3A%2F%2Fopulentyn.com-blue?style=flat&logo=google-chrome&logoColor=white" alt="Website">
+            </a>
+            &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="mailto:support@opulentyn.com">
+                <img src="https://img.shields.io/badge/Email-support%40opulentyn.com-blue?style=flat&logo=gmail&logoColor=white" alt="Email">
+            </a>
+            &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="https://wa.me/13435013587" target="_blank">
+                <img src="https://img.shields.io/badge/WhatsApp-%2B13435013587-blue?logo=whatsapp&logoColor=green" alt="WhatsApp">
+            </a>
+            &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="https://join.slack.com/t/opulentyn/shared_invite/zt-2s230jtbq-dWBs8XUZcrYim~nUqiimSA" target="_blank">
+                <img src="https://img.shields.io/badge/Slack-support--sdk-blueviolet?style=flat&logo=slack&logoColor=white" alt="Slack">
+            </a>
+        </div>
+        """
+    )
+
+    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("Process")
+        with gr.Column():
+            with gr.TabItem("Image"):
+                image_output = gr.Image(type="pil")
+            with gr.TabItem("Json"):
+                json_output = gr.JSON()
+    
+    process_button.click(process_image, inputs=[image_input], outputs=[image_output, json_output])
+
+    gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fopulentyn%2FLicensePlateRecognition"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fopulentyn%2FLicensePlateRecognition&countColor=%23263759" /></a>')
+
+demo.launch(server_name="0.0.0.0", server_port=7860)

install.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+sudo cp lib/libopyalprenv.so /usr/lib
+sudo cp lib/libtensorflow.so.2.6.0 /usr/lib
+sudo ln -s /usr/lib/libtensorflow.so.2.6.0 /usr/lib/libtensorflow.so.2
+sudo ln -s /usr/lib/libtensorflow.so.2.6.0 /usr/lib/libtensorflow.so.1
+sudo ln -s /usr/lib/libtensorflow.so.2.6.0 /usr/lib/libtensorflow.so
+
+sudo cp lib/libtensorflow_framework.so.2.6.0 /usr/lib
+sudo ln -s /usr/lib/libtensorflow_framework.so.2.6.0 /usr/lib/libtensorflow_framework.so.2
+sudo ln -s /usr/lib/libtensorflow_framework.so.2.6.0 /usr/lib/libtensorflow_framework.so
+

lib/libopymrzenv.so

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

license.txt

@@ -0,0 +1 @@
+ahdgG47dPO63awosnVCE6Q/H+Sk32sSJ7Lmvsxp9WoZiSdgej8wh+DjBQ2k2enKtNifOTGW+gqZVFhfcHXkVh4m3xSzQVWSmhUy06suXiH7bKJvzH0NdOY2yI2Mqn5gSqsqwhwQ4+HC5Ol4c3qAUJ0OwmEQx4qxYEofo3LK5uTc=

model/mrza.bin

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

model/mrzb.bin

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

model/params.bin

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

model/s.bin

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

opymrz.py

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

requirements.txt

@@ -0,0 +1,4 @@
+flask
+flask-cors
+gradio
+datadog_api_client

run.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+cd /home/opyalpr
+exec python3 demo.py &
+exec python3 app.py