INNER CODE UNIT · Python

generate_vertical_masks

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

def generate_vertical_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(0, 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
            masks.append(m)
    return masks


def generate_image_crops(image_path, image_size=224, pixel_size=10):

    assert pixel_size * 2 < image_size

    image = Image.open(image_path).convert("RGB")

View source record →