Add padding in readme

README.md

@@ -17,14 +17,35 @@ The example below demonstrates how to obtain an image embedding with the [AutoMo
 ```python
 import torch
 import cv2
+import numpy as np
 from torchvision import transforms
 from transformers import AutoImageProcessor, AutoModel
-from transformers.image_utils import load_image
+
+def padding(img):
+    """
+    :param img: take the image as input, np.uint8 format, [0-255] range
+    :return: img: square image with white pixel padded along the shorter side.
+    """
+    h, w, _ = img.shape
+    if h > w:
+        new_img = 255 * np.ones((h, h, 3)).astype(np.uint8)
+        start_w = int((h-w)/2)
+        new_img[:, start_w:start_w+w, :] = img
+        return new_img
+
+    elif h < w:
+        new_img = 255 * np.ones((w, w, 3)).astype(np.uint8)
+        start_h = int((w - h) / 2)
+        new_img[start_h:start_h + h, :, :] = img
+        return new_img
+    else:
+        return img
 
 image_path = "your local image path"
 dim = 384
 image = cv2.imread(image_path)
 image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+image = padding(image)
 image = cv2.resize(image, (dim, dim))
 transform = transforms.ToTensor()
 image = transform(image)