python - 检测黑白图像上的白色矩形并裁剪(OpenCV)
问题描述
我正在努力将图像裁剪为放置在图像内部的矩形大小。这是原始图像:
到目前为止,我能够输入它,将其颜色转换为 HSV 颜色空间并对其应用阈值。到目前为止,这是我的代码:
import cv2
#Read input image
img = cv2.imread('rdm_generated_image.png')
#convert from BGR to HSV color space
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
#get the saturation plane - all black/white/gray pixels are zero, and colored pixels are above zero.
s = hsv[:, :, 1]
#apply threshold on s
ret, thresh = cv2.threshold(s, 8, 255, cv2.THRESH_BINARY)
#invert colors, so every dark spots are now white
image = cv2.bitwise_not(thresh)
cv2.imwrite("image.png", image)
完成此操作后,程序将输出以下内容:
现在我只想将图像裁剪到中间的大框,但我无法检测到它的轮廓。
我尝试了它cv2.findContours(image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
和其他一些功能,但我还没有成功。如果这应该是错误的做法,请纠正我。感谢您的帮助,因为我对 OpenCV 有点缺乏经验。
提前致谢!
解决方案
这是在 Python/OpenCV 中执行此操作的一种方法。
- 阅读图片
- 转换为灰色
- 临界点
- 寻找轮廓
- 过滤大约预期区域
- 绘制轮廓
- 保存结果
输入:
import cv2
import numpy as np
#Read input image
img = cv2.imread('boxes.png')
#convert from BGR to HSV color space
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
#apply threshold
thresh = cv2.threshold(gray, 30, 255, cv2.THRESH_BINARY)[1]
# find contours and get one with area about 180*35
# draw all contours in green and accepted ones in red
contours = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
contours = contours[0] if len(contours) == 2 else contours[1]
#area_thresh = 0
min_area = 0.95*180*35
max_area = 1.05*180*35
result = img.copy()
for c in contours:
area = cv2.contourArea(c)
cv2.drawContours(result, [c], -1, (0, 255, 0), 1)
if area > min_area and area < max_area:
cv2.drawContours(result, [c], -1, (0, 0, 255), 1)
# save result
cv2.imwrite("box_found.png", result)
# show images
cv2.imshow("GRAY", gray)
cv2.imshow("THRESH", thresh)
cv2.imshow("RESULT", result)
cv2.waitKey(0)
结果:
推荐阅读
- typescript - 使属性装饰器类型安全似乎只适用于简单的类?
- python - 从 Pandas 的索引中获取类
- reactjs - github action npm run build 不工作,但在本地工作
- javascript - 如何将一个 HTML 表单输入发送到两个不同的 API 并从中获取结果?
- javascript - channel.send 不是函数
- flutter - 以编程方式关闭 Flutter 上的 GPS
- html - 为什么 chrome 开发工具在正文中插入额外的 div
- django - Wagtail 设置仅使用 .dev
- linux - 在将文件中的日期与系统日期进行比较时出现错误:第 7 行:10#2020-12-09: value too great for base (error token is "09")
- javascript - 如何在图像单击时打开图像模式