INNER CODE UNIT · Python

generate_image_crops

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

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")
    image = pad_to_square(image, size=image_size)
    gray = np.ones_like(image) * 128

    image_crops = [image]
    masks = [np.ones_like(image)]

    horizontal_masks = generate_horizontal_masks(image_size, pixel_size)
    vertical_masks = generate_vertical_masks(image_size, pixel_size)
    for m in horizontal_masks+vertical_masks:
        m_bar = 1-m
        m_masked = image * m + gray * m_bar
        m_masked = m_masked.astype(np.uint8)
        image_crops.append(Image.fromarray(m_masked))

View source record →