INNER CODE UNIT · Python

pad_to_square

patrickjohncyh/fashion-clip · fashion_clip/attention_map.py:8

def pad_to_square(image, size=224):
    original_size = image.size
    ratio = float(size) / max(original_size)
    new_size = tuple([int(x * ratio) for x in original_size])
    new_im = Image.new("RGB", size=(size, size), color=(128, 128, 128))
    new_im.paste(image.resize(new_size, Image.ANTIALIAS),
                 ((size - new_size[0]) // 2,
                  (size - new_size[1]) // 2))
    return new_im


def generate_horizontal_masks(image_size, pixel_size, channels=3):
    base_mask = np.ones((image_size, image_size, channels))
    masks = []
    n_pixels = image_size // pixel_size + 1
    for i in range(1, n_pixels):
        for j in range(i+1, n_pixels):
            m = base_mask.copy()

View source record →