使用Python从图片中查找彩色形状的数量

from PIL import Image
from colorthief import ColorThief

image = Image.open('image.jpg')

color_thief = ColorThief('image.jpg')
dominant_color = color_thief.get_color(quality=1)

width, height = image.size
block_size = 10  # 定义每个小块的大小
num_blocks = (width // block_size) * (height // block_size)  # 计算小块的数量

for i in range(0, width, block_size):
    for j in range(0, height, block_size):
        block = image.crop((i, j, i + block_size, j + block_size))
        # 在这里进行颜色判断和计数

target_color = (255, 0, 0)  # 定义目标颜色，这里以红色为例

def is_target_color(color):
    return color == target_color

count = 0  # 计数器，用于记录符合条件的小块数量

for i in range(0, width, block_size):
    for j in range(0, height, block_size):
        block = image.crop((i, j, i + block_size, j + block_size))
        pixels = list(block.getdata())
        if any(is_target_color(pixel) for pixel in pixels):
            count += 1

print("彩色形状的数量为：", count)