使用OpenCV创建包含在斑点内的矩形

Question

输入图片：

输出图片：

我在一张图片中有几个不同颜色的斑点，我想要在每个颜色的最大斑点内创建矩形（或正方形——这似乎更容易）。我已经找到了一个答案如何创建包含单个最大斑点的矩形，但是我不确定如何找到适合斑点的正方形。它不必是最大的，只需要比某个面积大就可以，否则我就不会包括它。我也看到了一些关于多边形的工作，但没有针对非规则形状的。

- Jacob M.

---

如果您提供一个未压缩的图片，我会在我的答案中同时发布您的图片上的结果。 - Miki

Answer 1

对于单个blob，该问题可以表述为：在一个矩阵中找到包含零元素的最大矩形。

要找到blob内最大轴向矩形，你可以参考我之前的回答中的findMinRect函数。该代码是Python原版此处的C++移植版本。

---

第二个问题是找到所有颜色相同的斑点。这有点棘手，因为您的图像是jpeg格式，压缩会在边界附近创建很多人工颜色。因此，我创建了一个png图像（如下所示），只是为了展示算法的工作原理。由您提供没有压缩伪影的图像。

然后，您只需要为每种颜色创建一个掩码，在该掩码中查找每个斑点的连通组件，并计算每个斑点的最小矩形。

初始图像：

这里展示了每个色块找到的矩形，按颜色划分。然后您可以只选择所需的矩形，即每种颜色的最大矩形或最大色块的矩形。

结果：

这是代码：

#include <opencv2/opencv.hpp>
#include <algorithm>
#include <set>
using namespace std;
using namespace cv;

// https://dev59.com/J3E95IYBdhLWcg3wDpmg#30418912
Rect findMinRect(const Mat1b& src)
{
    Mat1f W(src.rows, src.cols, float(0));
    Mat1f H(src.rows, src.cols, float(0));

    Rect maxRect(0, 0, 0, 0);
    float maxArea = 0.f;

    for (int r = 0; r < src.rows; ++r)
    {
        for (int c = 0; c < src.cols; ++c)
        {
            if (src(r, c) == 0)
            {
                H(r, c) = 1.f + ((r>0) ? H(r - 1, c) : 0);
                W(r, c) = 1.f + ((c>0) ? W(r, c - 1) : 0);
            }

            float minw = W(r, c);
            for (int h = 0; h < H(r, c); ++h)
            {
                minw = min(minw, W(r - h, c));
                float area = (h + 1) * minw;
                if (area > maxArea)
                {
                    maxArea = area;
                    maxRect = Rect(Point(c - minw + 1, r - h), Point(c + 1, r + 1));
                }
            }
        }
    }

    return maxRect;
}


struct lessVec3b
{
    bool operator()(const Vec3b& lhs, const Vec3b& rhs) {
        return (lhs[0] != rhs[0]) ? (lhs[0] < rhs[0]) : ((lhs[1] != rhs[1]) ? (lhs[1] < rhs[1]) : (lhs[2] < rhs[2]));
    }
};

int main()
{
    // Load image
    Mat3b img = imread("path_to_image");

    // Find unique colors
    set<Vec3b, lessVec3b> s(img.begin(), img.end());

    // Divide planes of original image
    vector<Mat1b> planes;
    split(img, planes);
    for (auto color : s)
    {
        // Create a mask with only pixels of the given color
        Mat1b mask(img.rows, img.cols, uchar(255));
        for (int i = 0; i < 3; ++i)
        {
            mask &= (planes[i] == color[i]);
        }

        // Find blobs
        vector<vector<Point>> contours;
        findContours(mask, contours, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);

        for (int i = 0; i < contours.size(); ++i)
        {
            // Create a mask for each single blob
            Mat1b maskSingleContour(img.rows, img.cols, uchar(0));
            drawContours(maskSingleContour, contours, i, Scalar(255), CV_FILLED);

            // Find minimum rect for each blob
            Rect box = findMinRect(~maskSingleContour);

            // Draw rect
            Scalar rectColor(color[1], color[2], color[0]);
            rectangle(img, box, rectColor, 2);
        }
    }

    imshow("Result", img);
    waitKey();

    return 0;
}

- Miki

Answer 2

您可以使用此代码来定位任意形状内最大正方形或矩形。虽然它是MATLAB而不是C++/OpenCV，但您可以轻松更改其源代码以适应您的需求。

要查找凸多边形内最大矩形，请查看此处（附有代码）。

- herohuyongtao