opencv_tutorials/python/code_047/opencv_047.py at master · HZHCoder1990/opencv_tutorials · GitHub

43 lines (35 loc) · 1.47 KB

import cv2 as cv

import numpy as np

def connected_components_stats_demo(src):

src = cv.GaussianBlur(src, (3, 3), 0)

gray = cv.cvtColor(src, cv.COLOR_BGR2GRAY)

ret, binary_ = cv.threshold(gray, 0, 255, cv.THRESH_BINARY | cv.THRESH_OTSU)

# 使用开运算去掉外部的噪声

kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))

binary = cv.morphologyEx(binary_, cv.MORPH_OPEN, kernel)

cv.imshow("binary", binary_)

num_labels, labels, stats, centers = cv.connectedComponentsWithStats(binary, connectivity=8, ltype=cv.CV_32S)

colors = []

for i in range(num_labels):

b = np.random.randint(0, 256)

g = np.random.randint(0, 256)

r = np.random.randint(0, 256)

colors.append((b, g, r))

colors[0] = (0, 0, 0)

image = np.copy(src)

for t in range(1, num_labels, 1):

x, y, w, h, area = stats[t]

cx, cy = centers[t]

# 标出中心位置

cv.circle(image, (np.int32(cx), np.int32(cy)), 2, (0, 255, 0), 2, 8, 0)

# 画出外接矩形

cv.rectangle(image, (x, y), (x+w, y+h), colors[t], 1, 8, 0)

cv.putText(image, "No." + str(t), (x, y), cv.FONT_HERSHEY_SIMPLEX, .5, (0, 0, 255), 1);

print("label index %d, area of the label : %d"%(t, area))

cv.imshow("colored labels", image)

cv.imwrite("labels.png", image)

print("total rice : ", num_labels - 1)

input = cv.imread("granule.png")

connected_components_stats_demo(input)

cv.waitKey(0)

cv.destroyAllWindows()

Expand file tree