update model

musimab · Apr 11, 2022 · db166a2 · db166a2
1 parent a1d103d
commit db166a2
Show file tree

Hide file tree

Showing 2 changed files with 35,716 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -134,7 +134,7 @@ def getBoxRegions(regions):
     parser = argparse.ArgumentParser(description = 'Identity Card Information Extractiion')
     parser.add_argument('--folder_name', default="images", type=str, help='folder that contain tc id images')
     parser.add_argument('--neighbor_box_distance', default = 50, type = float, help='Nearest box distance threshold')
-    parser.add_argument('--face_recognition',  default = "haar", type = str,   help='face detection algorithm')
+    parser.add_argument('--face_recognition',  default = "ssd", type = str,   help='face detection algorithm')
     parser.add_argument('--ocr_method',  default = "EasyOcr", type = str,   help='Type of ocr method for converting images to text')
     parser.add_argument('--rotation_interval', default = 60,   type = int, help='Face search interval for rotation matrix')
     args = parser.parse_args()
@@ -152,7 +152,7 @@ def getBoxRegions(regions):
 
 
     start = time.time()
-
+    end = 0
     for filename in sorted(os.listdir(Folder)):
 
         img = cv2.imread(os.path.join(Folder,filename))
@@ -202,6 +202,7 @@ def getBoxRegions(regions):
         for id, val in PersonInfo.items():
             print(id,':' ,val)
         print(" ")
+        end = time.time()
         utlis.displayMachedBoxes(final_img, new_bboxes)
 
         utlis.displayAllBoxes(final_img, bbox_coordinates)
@@ -220,7 +221,7 @@ def getBoxRegions(regions):
         #cv2.imwrite("predicted_mask.jpg", predicted_mask)
         plt.show()
 
-    end = time.time()
+
     print("Execution Time:", (end -start))