头歌 python 智慧棋盘——使用图像处理找圆

棋盘——使用图像处理找圆"/>

头歌 python 智慧棋盘——使用图像处理找圆

import cv2
import numpy as npthresholdBlack = 120 # RGB 都小于这个值、或者灰度图像的值小于这个值，认为是 黑色
thresholdWhite = 200 # RGB 都大于这个值、或者灰度图像的值大于这个值，认为是 白色def fnJudge(fnIn, fnOut):numBlack = 0 # 记录 黑子 数目numWhite = 0 # 记录 白子 数目# Begin-end之间，补充读入文件、转换为灰度图的代码#********** Begin *********#imgIn = cv2.imread(fnIn, 3)gray_img = cv2.cvtColor(imgIn, cv2.COLOR_BGR2GRAY)  # 将彩色图，转为灰度图#********** End **********#fnout = imgIncircles = cv2.HoughCircles(gray_img, cv2.HOUGH_GRADIENT, 1, minDist=80,param1=100, param2=35, minRadius=20, maxRadius=110)circles = np.uint16(np.around(circles))# Begin-end之间，补充显示，计算的代码#********** Begin *********#for idx, i in enumerate(circles[0,:]): # 请思考，为啥用 [0, :]    cv2.circle(imgIn,(i[0],i[1]), i[2],(0,255,0), 2) # 在圆心处用红色画一个小圆cv2.circle(imgIn,(i[0],i[1]), 2, (0,0,255), 3)# 在圆心处，写上这是找到的第几个圆cv2.putText(fnout, str(idx), (i[1],i[0]), cv2.FONT_HERSHEY_COMPLEX, 0.5, (255, 0, 0), 2)if(gray_img[i[1],i[0]] < thresholdBlack ) : # 灰度小于这个就是 黑色numBlack += 1if(gray_img[i[1],i[0]] > thresholdBlack ) :numWhite += 1#********** End **********#return (numBlack,numWhite)