python - 找到在opencv(python)中设置的白色像素的边界框

Question

我需要找到白色像素云的最小区域矩形 ( 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

我该怎么办？谢谢

Answer 1

minAreaRect 的 python 文档具有误导性。

使用:

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

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

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

另请参阅:Checking contour area in opencv using python