在矩形内随机生成不相交的相同圆形

使用python和opencv在矩形内随机生成不相交的相同圆形

import cv2
import numpy as np

# 生成随机圆
class Random_circle(object):
    def __init__(self,minnum,maxnum,height,width,radius):
        self.circle = np.random.randint(minnum,maxnum) #生成随机圆形个数
        self.h = height  #图像高
        self.w = width  #图像宽
        self.radius = radius  #圆的半径
        self.circle_pos = []  #存储圆心坐标
        self.img = self.create_img()  #生成图像
    
    #检测随机生成的圆心与之前生成的圆心距离
    def radius_detect(self,x,y):  
        if(self.circle_pos):  #排除第一个无法比较的圆心
            for pos in self.circle_pos:  #列举所有之前生成的圆心坐标
                if(((pos[0]-x)**2+(pos[1]-y)**2)**0.5>=self.radius*2): #比较两点距离与半径*2的大小
                    continue #若大于等于，则继续比较
                else:
                    return 1 #若小于，则继续循环生成随机圆心坐标
                return 0 #列举结束，跳出循环
        else:
            return 0 
    
    def create_img(self):
        background = np.ones((self.h,self.w),np.uint8) #生成黑色背景
        for num in range(0,self.circle):
            x = np.random.randint(self.radius,self.w-self.radius) #防止圆形跑到边界外
            y = np.random.randint(self.radius,self.h-self.radius)
            while(self.radius_detect(x,y)): #比较是否重合
                x = np.random.randint(self.radius,self.w-self.radius)
                y = np.random.randint(self.radius,self.h-self.radius)  
            self.circle_pos.append((x,y)) #将不重合的圆添加到圆心坐标中
            img = cv2.circle(background,(x,y),self.radius,(255,255,255),-1) #描绘所有圆       
        return img
    

if __name__ == "__main__":
    #随机生成10-50个圆,图像高600，宽900，圆的半径20
    rand = Random_circle(minnum=10,maxnum=50,height=600,width=900,radius=20) 
    circle = rand.circle #读取圆的个数
    img = rand.img #读取图像
    
    print(circle) #打印个数
    cv2.imshow("circle",img) #显示图像
    cv2.waitKey(0)
    cv2.destroyAllWindows()

结果如下图