OpenCV python 计算轮廓等效直径

OpenCV python 计算轮廓等效直径

处理图片[cc.jpg]

import cv2
import numpy as np


def get_contour(img):
    """获取连通域

    :param img: 输入图片
    :return: 最大连通域
    """
    # 灰度化, 二值化, 连通域分析
    img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    ret, img_bin = cv2.threshold(img_gray, 127, 255, cv2.THRESH_BINARY)

    img_contour, contours, hierarchy = cv2.findContours(img_bin, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)

    return contours[0]


def main():

    # 1.导入图片
    img_src = cv2.imread("cc.jpg")
    img_result = img_src.copy()

    # 2.获取连通域
    cont = get_contour(img_src)
    cv2.drawContours(img_result, cont, -1, (0, 0, 255), 2)

    # 3.获取轮廓面积
    cnt_area = cv2.contourArea(cont)

    # 4.计算等效直径
    equi_diameter = np.sqrt(4 * cnt_area / np.pi)
    print("轮廓面积=", cnt_area)
    print("等效直径=%.4f" % equi_diameter)

    cv2.circle(img_result, (100, 100), int(equi_diameter/2), (0, 0, 255), 1)

    # 5.显示结果
    cv2.imshow("img_result", img_result)

    cv2.waitKey()
    cv2.destroyAllWindows()


if __name__ == '__main__':
    main()

处理结果图片[img_result.jpg]