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

import cv2
import numpy as np

thresholdBlack = 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 = imgIn

    circles = 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 += 1
        if(gray_img[i[1],i[0]] > thresholdBlack ) :
            numWhite += 1

    #********** End **********#

    return (numBlack,numWhite)