Spaces:
Running
Running
| import sys | |
| sys.path.append('.') | |
| import os | |
| import base64 | |
| import json | |
| import handtool | |
| import cv2 | |
| import numpy as np | |
| from roi import get_roi, mat_to_bytes | |
| from flask import Flask, request, jsonify | |
| threshold = 0.15 | |
| licensePath = "license.txt" | |
| license = "" | |
| config = handtool.EncoderConfig(29, 5, 5, 10) | |
| encoder = handtool.create_encoder(config) | |
| machineCode = encoder.getMachineCode() | |
| print("\nmachineCode: ", machineCode.decode('utf-8')) | |
| # Get a specific environment variable by name | |
| license = os.environ.get("LICENSE") | |
| # Check if the variable exists | |
| if license is not None: | |
| print("Value of LICENSE:") | |
| else: | |
| license = "" | |
| try: | |
| with open(licensePath, 'r') as file: | |
| license = file.read().strip() | |
| except IOError as exc: | |
| print("failed to open license.txt: ", exc.errno) | |
| print("license: ", license) | |
| ret = encoder.setActivation(license.encode('utf-8')) | |
| print("\nactivation: ", ret) | |
| _ = encoder.init() | |
| print("\ninit: ", ret) | |
| app = Flask(__name__) | |
| def compare_palmprint(): | |
| result = "None" | |
| similarity = -1 | |
| palm1 = None | |
| palm2 = None | |
| file1 = request.files['file1'] | |
| file2 = request.files['file2'] | |
| print("---- 1") | |
| try: | |
| image1 = cv2.imdecode(np.frombuffer(file1.read(), np.uint8), cv2.IMREAD_COLOR) | |
| except: | |
| result = "Failed to open file1" | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| try: | |
| image2 = cv2.imdecode(np.frombuffer(file2.read(), np.uint8), cv2.IMREAD_COLOR) | |
| except: | |
| result = "Failed to open file2" | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| img1 = mat_to_bytes(image1) | |
| img2 = mat_to_bytes(image2) | |
| hand_type_1, x11, y11, x12, y12, detect_state_1 = encoder.detect_using_bytes(img1) | |
| hand_type_2, x21, y21, x22, y22, detect_state_2 = encoder.detect_using_bytes(img2) | |
| palm1 = {"hand_type": hand_type_1, "x1": x11, "y1": y11, "x2": x12, "y2": y12} | |
| palm2 = {"hand_type": hand_type_2, "x1": x21, "y1": y21, "x2": x22, "y2": y22} | |
| if hand_type_1 != hand_type_2: | |
| result = "Different hand" | |
| # print(f"\n 2 images are from the different hand\n similarity: 0.0") | |
| similarity = 0.0 | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| if detect_state_1 == 0 or detect_state_2 == 0: | |
| if ret != 0: | |
| result = "Activation failed" | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| # print(f"\n hand detection failed !\n plesae make sure that input hand image is valid or not.") | |
| result = "hand detection failed !\nPlesae make sure that input hand image is valid or not." | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| if detect_state_1 >= 2 or detect_state_2 >= 2: | |
| # print(f"\n multi-hand detected !\n plesae put one hand image, not multiple hand.") | |
| result = "multi-hand detected !\nPlesae try on image with one hand , not multiple hand." | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| roi1 = mat_to_bytes(get_roi(image1, hand_type_1, x11, y11, x12, y12)) | |
| roi2 = mat_to_bytes(get_roi(image2, hand_type_2, x21, y21, x22, y22)) | |
| one_palmprint_code = encoder.encode_using_bytes(roi1) | |
| another_palmprint_code = encoder.encode_using_bytes(roi2) | |
| score = one_palmprint_code.compare_to(another_palmprint_code) | |
| if score >= threshold: | |
| result = "Same hand" | |
| similarity = score | |
| # print(f"\n 2 images are from the same hand\n similarity: {score}") | |
| else: | |
| result = "Different hand" | |
| similarity = score | |
| # print(f"\n 2 images are from the different hand\n similarity: {score}") | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| def compare_palmprint_base64(): | |
| result = "None" | |
| similarity = -1 | |
| palm1 = None | |
| palm2 = None | |
| content = request.get_json() | |
| try: | |
| imageBase64_1 = content['base64_1'] | |
| image_data1 = base64.b64decode(imageBase64_1) | |
| np_array = np.frombuffer(image_data1, np.uint8) | |
| image1 = cv2.imdecode(np_array, cv2.IMREAD_COLOR) | |
| except: | |
| result = "Failed to open file1" | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| try: | |
| imageBase64_2 = content['base64_2'] | |
| image_data2 = base64.b64decode(imageBase64_2) | |
| np_array = np.frombuffer(image_data2, np.uint8) | |
| image2 = cv2.imdecode(np_array, cv2.IMREAD_COLOR) | |
| except IOError as exc: | |
| result = "Failed to open file2" | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| img1 = mat_to_bytes(image1) | |
| img2 = mat_to_bytes(image2) | |
| hand_type_1, x11, y11, x12, y12, detect_state_1 = encoder.detect_using_bytes(img1) | |
| hand_type_2, x21, y21, x22, y22, detect_state_2 = encoder.detect_using_bytes(img2) | |
| palm1 = {"hand_type": hand_type_1, "x1": x11, "y1": y11, "x2": x12, "y2": y12} | |
| palm2 = {"hand_type": hand_type_2, "x1": x21, "y1": y21, "x2": x22, "y2": y22} | |
| if hand_type_1 != hand_type_2: | |
| result = "Different hand" | |
| # print(f"\n 2 images are from the different hand\n similarity: 0.0") | |
| similarity = 0.0 | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| if detect_state_1 == 0 or detect_state_2 == 0: | |
| if ret != 0: | |
| result = "Activation failed" | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| # print(f"\n hand detection failed !\n plesae make sure that input hand image is valid or not.") | |
| result = "hand detection failed !\nPlesae make sure that input hand image is valid or not." | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| if detect_state_1 >= 2 or detect_state_2 >= 2: | |
| # print(f"\n multi-hand detected !\n plesae put one hand image, not multiple hand.") | |
| result = "multi-hand detected !\nPlesae try on image with one hand , not multiple hand." | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| roi1 = mat_to_bytes(get_roi(image1, hand_type_1, x11, y11, x12, y12)) | |
| roi2 = mat_to_bytes(get_roi(image2, hand_type_2, x21, y21, x22, y22)) | |
| one_palmprint_code = encoder.encode_using_bytes(roi1) | |
| another_palmprint_code = encoder.encode_using_bytes(roi2) | |
| score = one_palmprint_code.compare_to(another_palmprint_code) | |
| if score >= threshold: | |
| result = "Same hand" | |
| similarity = score | |
| # print(f"\n 2 images are from the same hand\n similarity: {score}") | |
| else: | |
| result = "Different hand" | |
| similarity = score | |
| # print(f"\n 2 images are from the different hand\n similarity: {score}") | |
| response = jsonify({"compare_result": result, "compare_similarity": similarity, "palm1": palm1, "palm2": palm2}) | |
| response.status_code = 200 | |
| response.headers["Content-Type"] = "application/json; charset=utf-8" | |
| return response | |
| if __name__ == '__main__': | |
| port = int(os.environ.get("PORT", 8080)) | |
| app.run(host='0.0.0.0', port=port) |