opencv: 绘制矩形轮廓框,并记录轮廓框坐标

步骤简述

使用OpenCV绘制矩形轮廓框,一般包括如下步骤:

  1. 转换为灰度图;
  2. 进行阈值处理;
  3. 进行中值滤波;
  4. 在原始图像上绘制矩形框。

附图解析

原始图像:
opencv: 绘制矩形轮廓框,并记录轮廓框坐标_第1张图片

第一步,转换为灰度图:
opencv: 绘制矩形轮廓框,并记录轮廓框坐标_第2张图片

第二步,经过阈值处理:
opencv: 绘制矩形轮廓框,并记录轮廓框坐标_第3张图片

第三步,中值滤波后:
opencv: 绘制矩形轮廓框,并记录轮廓框坐标_第4张图片

最后一步,在原始图像上绘制矩形框:
opencv: 绘制矩形轮廓框,并记录轮廓框坐标_第5张图片

生成的记录文件(矩形轮廓框四个端点的平面坐标)

1: [367,201  367,210  371,201  371,210]
2: [362,195  362,197  363,195  363,197]
3: [297,187  297,190  301,187  301,190]
4: [396,182  396,184  398,182  398,184]
5: [371,168  371,190  394,168  394,190]
6: [382,166  382,169  387,166  387,169]
7: [211,160  211,189  238,160  238,189]
8: [377,148  377,163  402,148  402,163]
9: [308,148  308,217  353,148  353,217]
10: [315,193  315,215  348,193  348,215]
11: [277,148  277,188  311,148  311,188]
12: [241,147  241,190  274,147  274,190]
13: [179,132  179,189  235,132  235,189]
14: [355,129  355,192  372,129  372,192]

Code

附上自己写的实验代码:

# coding=utf-8
import cv2

# 原图像路径
origin_pic = cv2.imread('./origin.jpg')
# 文档路径,用于记录轮廓框坐标
txt_file = open('./contours.txt', 'w')

# 要先转换成单通道灰度图像才能进行后续的图像处理
pic = cv2.cvtColor(origin_pic, cv2.COLOR_BGR2GRAY)
# 阈值处理,将前景全填充为白色,背景全填充为黑色
_, pic = cv2.threshold(src=pic, thresh=200, maxval=255, type=1)
# 中值滤波,去除椒盐噪声
pic = cv2.medianBlur(pic, 5)
# 边缘检测,得到的轮廓列表
_1, contours, _2 = cv2.findContours(pic, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# 根据轮廓列表,循环在原始图像上绘制矩形边界
for i in range(len(contours)):
    cnt = contours[i]
    x, y, w, h = cv2.boundingRect(cnt)
    origin_pic = cv2.rectangle(origin_pic, (x, y), (x+w, y+h), (255, 0, 0), 2)
    txt_file.write('{}: [{},{}  {},{}  {},{}  {},{}]\n'.format(i+1, x, y, x, y+h, x+w, y, x+w, y+h))

cv2.imwrite('./rectangle.jpg', origin_pic)
txt_file.close()

cv2.imshow('', origin_pic)
cv2.waitKey(0)
cv2.destroyAllWindows()

你可能感兴趣的:(OpenCV,OpenCV-Python)