python+openCV图像处理（二十四）对象检测

import cv2 as cv
import numpy as np

def measure_object(image):
    gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)
    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY_INV | cv.THRESH_OTSU)
    print("threshold value : %s"%ret)
    cv.imshow("binary image", binary)
    dst = cv.cvtColor(binary, cv.COLOR_GRAY2BGR)
    outImage, contours, hireachy = cv.findContours(binary, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)
    print("--------------------------------")

    for i, contour in enumerate(contours):
        area = cv.contourArea(contour)#轮廓面积
        x, y, w, h = cv.boundingRect(contour)#外接矩形
        rate = min(w, h)/max(w, h)#宽高比
        print("rectangle rate : %s"%rate)

        mm = cv.moments(contour)#计算图像中的中心矩
        #print(type(mm))
        #重心坐标xc,yc
        cx = mm['m10']/mm['m00']
        cy = mm['m01']/mm['m00']
        cv.circle(dst, (np.int(cx), np.int(cy)), 3, (0, 255, 255), -1)
        cv.rectangle(dst, (x, y), (x+w, y+h), (0, 0, 255), 2)
        print("contour area ：%s"%area)
        print("--------------------------------")

        cv.drawContours(image, contours, i, (0, 255, 0), 2)
    cv.imshow("measure-contours", image)

def measure_object1(image):
    gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)
    ret, binary = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_OTSU)
    print("threshold value : %s"%ret)
    cv.imshow("binary image1", binary)
    dst = cv.cvtColor(binary, cv.COLOR_GRAY2BGR)
    outImage, contours, hireachy = cv.findContours(binary, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)
    print("--------------------------------")

    for i, contour in enumerate(contours):
        area = cv.contourArea(contour)#轮廓面积
        x, y, w, h = cv.boundingRect(contour)#外接矩形
        rate = min(w, h)/max(w, h)#宽高比
        print("rectangle rate : %s"%rate)

        mm = cv.moments(contour)#计算图像中的中心矩
        #print(type(mm))
        #重心坐标xc,yc
        cx = mm['m10']/mm['m00']
        cy = mm['m01']/mm['m00']
        cv.circle(dst, (np.int(cx), np.int(cy)), 3, (0, 255, 255), -1)
        cv.rectangle(dst, (x, y), (x+w, y+h), (0, 0, 255), 2)
        print("contour area ：%s"%area)
        print("--------------------------------")

        approxCurve = cv.approxPolyDP(contour, 4, True)
        #print(approxCurve.shape)
        if approxCurve.shape[0] > 6:#圆
            cv.drawContours(image, contours, i, (0, 255, 0), 2)
        if approxCurve.shape[0] == 4:
            cv.drawContours(image, contours, i, (0, 0, 255), 2)
        if approxCurve.shape[0] == 3:#三角形
            cv.drawContours(image, contours, i, (255, 0, 0), 2)
    cv.imshow("measure-contours1", image)

src = cv.imread("0.png")
src1 = cv.imread("1.png")

cv.namedWindow("input image", cv.WINDOW_AUTOSIZE)
cv.imshow("input image", src)
cv.imshow("input image1", src1)
src2=src.copy()
src3=src1.copy()

measure_object(src2)
measure_object1(src3)
cv.waitKey(0)

cv.destroyAllWindows()