import cv2 import numpy as np import random img = np.ones((512, 512, 3), dtype=np.uint8)*255 def create_random_shape(): # 随机选择形状类型：0为矩形，1为三角形 shape_type = random.randint(0, 1) # 随机生成颜色 color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)) # 随机生成形状的起始坐标 x1 = random.randint(0, 500) y1 = random.randint(0, 500) # 随机生成形状的宽和高 width = random.randint(10, 100) height = random.randint(10, 100) if shape_type == 0: # 绘制矩形 x2 = x1 + width y2 = y1 + height cv2.rectangle(img, (x1, y1), (x2, y2), color, -1) else: # 绘制三角形 x2 = x1 + width x3 = random.randint(x1, x2) y2 = y1 + height y3 = y1 points = np.array([(x1, y1), (x2, y2), (x3, y3)]) cv2.drawContours(img, [points], 0, color, -1) for i in range(0, 10): create_random_shape() cv2.imshow("Random Shapes", img) cv2.waitKey(0) cv2.destroyAllWindows()我想让这段代码生成三角形和矩形的同时也生成圆，然后提取出来他们的区域

可以将代码中的 `shape_type` 添加一个选项，用于选择绘制圆形：

def create_random_shape():
    # 随机选择形状类型：0为矩形，1为三角形，2为圆形
    shape_type = random.randint(0, 2)
    
    # 随机生成颜色
    color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
    
    # 随机生成形状的起始坐标
    x1 = random.randint(0, 500)
    y1 = random.randint(0, 500)
    
    # 随机生成形状的宽和高
    width = random.randint(10, 100)
    height = random.randint(10, 100)
    
    if shape_type == 0:  # 绘制矩形
        x2 = x1 + width
        y2 = y1 + height
        cv2.rectangle(img, (x1, y1), (x2, y2), color, -1)
    elif shape_type == 1:  # 绘制三角形
        x2 = x1 + width
        x3 = random.randint(x1, x2)
        y2 = y1 + height
        y3 = y1
        points = np.array([(x1, y1), (x2, y2), (x3, y3)])
        cv2.drawContours(img, [points], 0, color, -1)
    else:  # 绘制圆形
        radius = random.randint(10, 50)
        cv2.circle(img, (x1, y1), radius, color, -1)

提取出不同形状的区域可以通过图像处理的方法来实现。例如，可以通过颜色来区分不同形状，然后使用形态学操作提取出不同形状的区域。具体实现方式可以参考 OpenCV 的相关文档和示例代码。