1

我需要找到一团白色像素的最小区域矩形（ cv2.minAreaRect() ）。

我的图像中有多个白色物体，我想在它们周围画一个矩形。

我在 C++ 中找到了解决方案：

cv::cvtColor(img, gray, CV_BGR2GRAY);
std::vector<cv::Point> points;
cv::Mat_<uchar>::iterator it = gray.begin<uchar>();
cv::Mat_<uchar>::iterator end = gray.end<uchar>();
for (; it != end; ++it)
{
    if (*it) points.push_back(it.pos());
}
cv::RotatedRect box = cv::minAreaRect(cv::Mat(points));

这是我在 python 中的尝试：

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray, (5 ,5), 0)
retval, thresh = cv2.threshold(gray, 210, 255, cv2.THRESH_BINARY)
whitep = []
for y, row in enumerate(thresh):
    for x, px in enumerate(row):
        if px == 255:
            whitep.append((y, x))
box = cv2.minAreaRect(whitep)

但它不起作用：

box = cv2.minAreaRect(whitep)
TypeError: <unknown> is not a numpy array

我能怎么做？谢谢

4

2 回答 2

4

python 文档minAreaRect具有误导性。

采用：

box = cv2.minAreaRect(numpy.array([whitep], dtype=numpy.int32))

这会将形状 (1,N,2) 的数组传递给 minAreaRect。

如果您使用默认整数类型为 numpy.int64 的系统，则需要指定 dtype。明确表达是最安全的。

另见：使用python检查opencv中的轮廓区域

于 2013-01-19T16:31:15.780 回答
1

你也可以试试这个。minAreaRect 在某些情况下确实失败了，但下面所述的方法总是有效。虽然使用 PIL。

gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray, (5 ,5), 0)
retval,thresh = cv2.threshold(gray, 210, 255, cv2.THRESH_BINARY)
mask=Image.fromarray(thresh)
box = mask.getbbox()
于 2020-09-25T03:37:34.877 回答