ykcheungab
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
#%%
|
|
import numpy as np
|
|
import cv2
|
|
|
|
|
|
def generate_checker_pattern(height, width, patch_size):
|
|
"""
|
|
Generate a black-and-white checkerboard pattern.
|
|
|
|
Each `patch_size x patch_size` block alternates between black and white.
|
|
Partial blocks at the image borders are also filled correctly.
|
|
"""
|
|
pattern = np.zeros((height, width, 3), dtype=np.uint8)
|
|
|
|
num_patches_x = (height + patch_size - 1) // patch_size
|
|
num_patches_y = (width + patch_size - 1) // patch_size
|
|
|
|
for i in range(num_patches_x):
|
|
for j in range(num_patches_y):
|
|
color = 255 if (i + j) % 2 == 0 else 0
|
|
y0 = i * patch_size
|
|
y1 = min((i + 1) * patch_size, height)
|
|
x0 = j * patch_size
|
|
x1 = min((j + 1) * patch_size, width)
|
|
pattern[y0:y1, x0:x1] = color
|
|
print("Patch ({}, {}) -> {}".format(i, j, "white" if color == 255 else "black"))
|
|
|
|
return pattern
|
|
|
|
|
|
def main():
|
|
height = 1200
|
|
width = 1000
|
|
patch_size = 12
|
|
|
|
pattern = generate_checker_pattern(height, width, patch_size)
|
|
|
|
cv2.imshow("Checker Pattern", pattern)
|
|
cv2.waitKey(0)
|
|
cv2.destroyAllWindows()
|
|
|
|
cv2.imwrite("checker_pattern.png", pattern)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|