利用opecv生成一些不规则mask

import os
import random

print(os.path.dirname(os.path.abspath(__file__)))

import cv2
import numpy as np

# 设置随机种子以获得可重复的结果
np.random.seed(0)

# 生成随机多边形顶点
num_vertices = random.randint(15,50) # 多边形顶点数量
height, width = 500, 500 # 图像大小
x_coords = np.random.randint(0, width, num_vertices)
y_coords = np.random.randint(0, height, num_vertices)
pts = np.array([[x, y] for x, y in zip(x_coords, y_coords)], np.int32)
pts = pts.reshape((-1, 1, 2))

image = np.zeros((height, width), np.uint8)

# 绘制多边形
cv2.polylines(image, [pts], True, (255, 255, 255), 2)
c_max = []
max_area = 0
max_cnt = 0
contours, hierarchy = cv2.findContours(image, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
for i in range(len(contours)):
cnt = contours[i]
area = cv2.contourArea(cnt)
if (area>max_area):
if(max_area!=0):
c_min = []
c_min.append(max_cnt)
cv2.drawContours(image, c_min, -1, (0,0,0), cv2.FILLED)
max_area = area
max_cnt = cnt
else:
c_min = []
c_min.append(cnt)
cv2.drawContours(image, c_min, -1, (0,0,0), cv2.FILLED)
# 显示图像
c_max.append(max_cnt)

cv2.drawContours(image, c_max, -1, (255, 255, 255), thickness=-1)

cv2.imshow('Irregular Shape', image)
cv2.waitKey(0)
cv2.destroyAllWindows()