INNER CODE UNIT · Python
n_pixels
patrickjohncyh/fashion-clip · fashion_clip/attention_map.py:22
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
masks.append(m)
return masks
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