在图像中检测圆并指定圆心和半径的图像区域ROI(抠图)。

在图像中检测圆并指定圆心和半径的图像区域ROI(抠图)。

# 加载环境包
import cv2
import numpy as np
import scipy.misc
import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator
# 图片简单处理
img = cv2.imread('G:/Ubuntu/Pythonlianx/30.0.jpg')  # 读取图片
GrayImage = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)  # 灰度化
GrayImage = cv2.medianBlur(GrayImage, 5)  # 中值模糊

# 阈值处理，输入图片默认为单通道灰度图片
ret, th1 = cv2.threshold(GrayImage, 127, 255, cv2.THRESH_BINARY)  # 固定阈值二值化
# threshold为固定阈值二值化
# 第二参数为阈值
# 第三参数为当像素值超过了阈值（或者小于阈值，根据type来决定），所赋予的值（一般情况下，都是256色，所以默认最大为255）
# thresh_binary是基于直方图的二值化操作类型，配合threshold一起使用。此外还有cv2.THRESH_BINARY； cv2.THRESH_BINARY_INV；
# cv2.THRESH_TRUNC； cv2.THRESH_TOZERO；cv2.THRESH_TOZERO_INV
th2 = cv2.adaptiveThreshold(GrayImage, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 3, 5)
# adaptiveThreshold自适应阈值二值化，自适应阈值二值化函数根据图片一小块区域的值来计算对应区域的阈值，从而得到也许更为合适的图片。
# 第二参数为当像素值超过了阈值（或者小于阈值，根据type来决定），所赋予的值（一般情况下，都是256色，所以默认最大为255）
# 第三参数为阈值计算方法，类型有cv2.ADAPTIVE_THRESH_MEAN_C，cv2.ADAPTIVE_THRESH_GAUSSIAN_C
# 第四参数是基于直方图的二值化操作类型，配合threshold一起使用。此外还有cv2.THRESH_BINARY； cv2.THRESH_BINARY_INV；
# cv2.THRESH_TRUNC； cv2.THRESH_TOZERO；cv2.THRESH_TOZERO_INV
# 第五参数是图片中分块的大小
# 第六参数是阈值计算方法中的常数项
th3 = cv2.adaptiveThreshold(GrayImage, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY, 3, 5)
# 同上
kernel = np.ones((5, 5), np.uint8)  # 创建全一矩阵，数值类型设置为uint8
erosion = cv2.erode(th2, kernel, iterations=1)  # 腐蚀处理
dilation = cv2.dilate(erosion, kernel, iterations=1)  # 膨胀处理

imgray = cv2.Canny(erosion, 20, 50)  # Canny算子边缘检测

circles = cv2.HoughCircles(imgray, cv2.HOUGH_GRADIENT, 1, 100, param1=100, param2=30, minRadius=100,
                           maxRadius=170)  # 霍夫圆变换
# 第3参数默认为1
# 第4参数表示圆心与圆心之间的距离（太大的话，会很多圆被认为是一个圆）
# 第5参数默认为100
# 第6参数根据圆大小设置(圆越小设置越小，检测的圆越多，但检测大圆会有噪点)
# 第7圆最小半径
# 第8圆最大半径
circles = np.uint16(np.around(circles))
# np.uint16数组转换为16位，0-65535
# np.around返回四舍五入后的值

P = circles[0]  # 去掉circles数组一层外括号
#for i in P:
    # 画出外圆
#    cv2.circle(img, (i[0], i[1]), i[2], (0, 255, 0), 2)
    # 第二参数（）内是圆心坐标，第三参数是半径，第四参数（）内是颜色，第五参数是线条粗细
    # 画出圆心
#    cv2.circle(img, (i[0], i[1]), 2, (0, 0, 255), 3)
print("圆的个数是：")
print(len(P))
for i in P:
    r = int(i[2])
    x = int(i[0])
    y = int(i[1])
    print("圆心坐标为：", (x, y))
    print("圆的半径是：", r)
shp=img.shape
print(shp)
circle = np.zeros(img.shape[0:2], dtype=np.uint8)
cv2.circle(circle, (645, 480), 135, 255, -1)
#遍历替换
sss=np.zeros(img.shape[:2],dtype=np.uint8)
sss = cv2.cvtColor(sss, cv2.COLOR_GRAY2BGR)
print(sss.shape)
rows,cols = circle.shape
center=[0,0]
for i in range(rows):
    for j in range(cols):
        if circle[i,j]==255: #255代表白色的点
            sss[center[0]+i,center[1]+j]=img[i,j]#此处替换颜色，为BGR通道
sss=sss[280:680,445:845]#裁剪坐标为[y0:y1,x0:x1]
print(sss.shape)

cv2.imwrite('30.jpg', sss)

#plt1 = plt.subplot(131)
#cv2.imshow('detected circles', img) #通过fig添加子图，参数：行数，列数，第几个。
#plt2 = plt.subplot(132)
#cv2.imshow('m',circle)
#plt3 = plt.subplot(133)
cv2.imshow('mask',sss)
cv2.waitKey(0)  # 无穷大等待时间
cv2.destroyAllWindows()