Update passport_img.py

passport_img.py

@@ -4,12 +4,16 @@ import matplotlib.pyplot as plt
 
 
 
-def crop_passport_size(img_arr):
+def crop_passport_size(img_arr,size):
+    if size==3545:
+        passport_size = (310, 410)
+    elif size==34:
+        passport_size = (260, 346)
     # Load the input image
     # original_image = cv2.imread(input_image_path)
     original_image = img_arr
     # Define the dimensions for the passport-size photo (35mm x 45mm)
-    passport_size = (310, 410)
+    # passport_size = (310, 410)
 
     # Resize the image to the passport-size dimensions
     resized_image = cv2.resize(original_image, (passport_size[0], passport_size[1]))
@@ -23,54 +27,71 @@ def crop_passport_size(img_arr):
 def add_border(image, top, bottom, left, right, color):
     return cv2.copyMakeBorder(image, top, bottom, left, right, cv2.BORDER_CONSTANT, value=color)
 
-def horizontal_merge(number,crop_img):
-    white_image = np.ones((450, 350, 3), dtype=np.uint8) * 255
+def horizontal_merge(number,crop_img,size):
+    if size==3545:
+        white_image = np.ones((450, 350, 3), dtype=np.uint8) * 255
+        img_per_row = 6
+    elif size==34:
+        white_image = np.ones((386, 300, 3), dtype=np.uint8) * 255
+        img_per_row = 7
+    # white_image = np.ones((450, 350, 3), dtype=np.uint8) * 255
     list_img = []
     # for i in range(5):
     for i in range(number):
         list_img.append(crop_img)
-    for i in range(6-number):
+    for i in range(img_per_row-number):
         list_img.append(white_image)
     merge_img = cv2.hconcat(list_img)
     return merge_img
 
-def merge_all(number,input_path):
+def merge_all(number,input_path,size):
     
     input_image_path = input_path  # Replace with the path to your input image
     # output_image_path = output_path  # Replace with the desired output path
-    output=crop_passport_size(input_image_path)
+    output=crop_passport_size(input_image_path,size)
 
     w_output = add_border(output, 4, 4, 4, 4, (0,0,0))
     w_output = add_border(w_output, 15, 15, 15, 15, (255,255,255))
     w_output = add_border(w_output, 1, 1, 1, 1, (0,0,0))
 
+    if size==3545:
+        img_per_row = 6
+        bottom_part = np.ones((270, 2100, 3), dtype=np.uint8) * 255
+    elif size==34:
+        img_per_row = 7
+        bottom_part = np.ones((268, 2100, 3), dtype=np.uint8) * 255
+
     img_2 = []
     num_ = number
-    if num_%6 == 0:
-        for i in range(int(num_/6)):
-            img1 = horizontal_merge(6,w_output)
+    if num_%img_per_row == 0:
+        for i in range(int(num_/img_per_row)):
+            img1 = horizontal_merge(img_per_row,w_output,size)
             img_2.append(img1)
     #         print('fill')
-        for i in range(6-int(num_/6)):
-            img1_1 = horizontal_merge(0,w_output)
+        for i in range(img_per_row-int(num_/img_per_row)):
+            img1_1 = horizontal_merge(0,w_output,size)
             img_2.append(img1_1)
     #         print('blank')
-    elif num_%6 != 0:
-        for i in range(int(num_//6)):
-            img1 = horizontal_merge(6,w_output)
+    elif num_%img_per_row != 0:
+        for i in range(int(num_//img_per_row)):
+            img1 = horizontal_merge(img_per_row,w_output,size)
             img_2.append(img1)
     #         print('fill')
-        img1 = horizontal_merge(num_%6,w_output)
+        img1 = horizontal_merge(num_%img_per_row,w_output,size)
         img_2.append(img1)
     #     print('some')
-        for i in range(6-int(num_//6)-1):
-            img1 = horizontal_merge(0,w_output)
+        for i in range(img_per_row-int(num_//img_per_row)-1):
+            img1 = horizontal_merge(0,w_output,size)
             img_2.append(img1)
     #         print('blank')
+
     merge_img_final = cv2.vconcat(img_2)
-    bottom_part = np.ones((270, 2100, 3), dtype=np.uint8) * 255
+    # print(merge_img_final.shape)
+    # bottom_part = np.ones((270, 2100, 3), dtype=np.uint8) * 255
     merge_img_final = cv2.vconcat([merge_img_final,bottom_part])
-    merge_img_final = add_border(merge_img_final, 40, 20, 30, 30, (255,255,255))
+    # print(merge_img_final.shape)
+    merge_img_final = add_border(merge_img_final, 52, 32, 30, 30, (255,255,255))
+    # print(merge_img_final.shape)
     # cv2.imwrite(output_path, merge_img_final)
     # print(merge_img_final.shape)
     return merge_img_final