Update passport_img.py

passport_img.py

@@ -15,6 +15,8 @@ def crop_passport_size(img_arr,size):
         passport_size = (310, 410)
     elif size==34:
         passport_size = (260, 346)
+    else:
+        passport_size = (170, 210)
     # Load the input image
     # original_image = cv2.imread(input_image_path)
     original_image = img_arr
@@ -40,6 +42,9 @@ def horizontal_merge(number,crop_img,size):
     elif size==34:
         white_image = np.ones((386, 300, 3), dtype=np.uint8) * 255
         img_per_row = 7
+    else:
+        white_image = np.ones((250, 210, 3), dtype=np.uint8) * 255
+        img_per_row = 10
     # white_image = np.ones((450, 350, 3), dtype=np.uint8) * 255
     list_img = []
     # for i in range(5):
@@ -66,6 +71,9 @@ def merge_all(number,input_path,size):
     elif size==34:
         img_per_row = 7
         bottom_part = np.ones((268, 2100, 3), dtype=np.uint8) * 255
+    else:
+        img_per_row = 10
+        bottom_part = np.ones((470, 2100, 3), dtype=np.uint8) * 255
 
     img_2 = []
     num_ = number