Upload 2 files

demo.py

@@ -61,7 +61,7 @@ def get_face_encoding(path):
 #     if cv2.waitKey(1) & 0xFF == ord('q'):
 #         break
 
-
+# frames are the snapshot of the video
 def process_frame(frame, process_this_frame, face_locations, faces, face_names, score):
 
     hkid_face_encoding = get_face_encoding("image")
@@ -84,12 +84,12 @@ def process_frame(frame, process_this_frame, face_locations, faces, face_names,
         #     print(f'frame: {len(frame)}')
             # try: 
             #     small_frame = cv2.imread(image_dir)
-        # small_frame = cv2.resize(frame, (0, 0), fx=0.25, fy=0.25)
+        small_frame = cv2.resize(frame, (0, 0), fx=0.25, fy=0.25)
         # else:
         #     print('fram has nth')
 
         # Convert the image from BGR color (which OpenCV uses) to RGB color (which face_recognition uses)
-        rgb_small_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # edited
+        rgb_small_frame = cv2.cvtColor(small_frame, cv2.COLOR_BGR2RGB)
         
         # Find all the faces and face encodings in the current frame of video
         face_locations = face_recognition.face_locations(rgb_small_frame)
@@ -142,10 +142,10 @@ def process_frame(frame, process_this_frame, face_locations, faces, face_names,
         
     for (top, right, bottom, left), name in zip(face_locations, face_names):
         # Scale back up face locations since the frame we detected in was scaled to 1/4 size
-        # top *= 4
-        # right *= 4
-        # bottom *= 4
-        # left *= 4
+        top *= 4
+        right *= 4
+        bottom *= 4
+        left *= 4
 
         # Draw a box around the face
         cv2.rectangle(frame, (left, top), (right, bottom), (65, 181, 41), 4)

similarity_check.py

@@ -1,10 +1,10 @@
-from model1 import model1
-from model2 import model2
+from model0 import model0
 import checkTool as ct
 import extract_pdf as pf
 import extraction_data as ed
 import get_chinese_name as cn
 import search_engine as se
+import get_chinese_code as cc
 
 # get info from hkid card
 
@@ -43,29 +43,19 @@ def get_data(img1_path, img2_path):
 
     # img_fp = 'IMG_4495.jpg'
 
-    info1 = model1(img1_path)
-    info2 = model2(img1_path)
+    # info1 = model1(img1_path)
+    # info2 = model2(img1_path)
 
-    def print_info(name, valid_hkid, hkid, issuedate):
-        print(f'Name: {name}') # name is without space
-        print(f'HKID: {hkid} and validity: {valid_hkid}')
-        print(f'Date of issue: {issuedate}')
+    # def print_info(name, valid_hkid, hkid, issuedate):
+    #     print(f'Name: {name}') # name is without space
+    #     print(f'HKID: {hkid} and validity: {valid_hkid}')
+    #     print(f'Date of issue: {issuedate}')
 
-    cinfo = ct.combine_info(info1, info2)
+    # cinfo = ct.combine_info(info1, info2)
+    cinfo = model0(img1_path)
+    print(cinfo)
 
     # get info from bank
-
-    # images = r'hangseng_page-0001.jpg'
-    # bank_list = ['bankofchina','hangsengbank','hsbc','sc']
-    # image_path = 'hangseng_page-0001.jpg'
-    # post_url = r''
-
-    # name = pf.get_info_from_bank(img2_path)
-    # name = pf.check_mr(name)
-    # name = name.replace(' ', '')
-    # name = name.lower()
-
-    # data = pf.get_info_from_bank(img2_path, file_name)
     data = ed.get_info_from_bank(img2_path)
     name = data["nameStatement"]
 
@@ -75,24 +65,17 @@ def get_data(img1_path, img2_path):
     # img_fp = 'IMG_1234.jpg'
     name1 = cinfo[0]
     threshold = 85
-    # print(f'Name in HKID: {name1}')
-    # print(f'Nmae in bank statement: {name}')
     similarity_score = string_similarity(name,name1)
-    # print(f'Similarity: {similarity_score}')
-    # if (similarity_score >= threshold): # Above threshold
-    #     print('It is the same person')
-    # else: # Below threshold
-    #     print('It is not the same person')
     data["similarity_score"] = similarity_score
     data["name_on_id"] = name1
     data["hkid"] = cinfo[2]
     data["validity"] = cinfo[1]
     data["issue_date"] = cinfo[3]
+    data["dateofbirth"] = cinfo[4]
 
-    # Search Engine
+    # Get chinese name
+    chi_name = cc.get_chinese_name(img1_path)
     # chi_name = cn.get_chiname(img1_path)["Chinese Name"]
-    chi_name = "allen lau"
     data["chi_name_id"] = chi_name
     
-    return data
-        
+    return data