INNER CODE UNIT · Python

ratio

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

    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()
            m[:min(i*pixel_size, image_size), :] = 0
            m[min(j*pixel_size, image_size):, :] = 0

View source record →