输入图片:
输出图片:
我在一张图片中有几个不同颜色的斑点,我想要在每个颜色的最大斑点内创建矩形(或正方形——这似乎更容易)。我已经找到了一个答案如何创建包含单个最大斑点的矩形,但是我不确定如何找到适合斑点的正方形。它不必是最大的,只需要比某个面积大就可以,否则我就不会包括它。我也看到了一些关于多边形的工作,但没有针对非规则形状的。
输入图片:
输出图片:
我在一张图片中有几个不同颜色的斑点,我想要在每个颜色的最大斑点内创建矩形(或正方形——这似乎更容易)。我已经找到了一个答案如何创建包含单个最大斑点的矩形,但是我不确定如何找到适合斑点的正方形。它不必是最大的,只需要比某个面积大就可以,否则我就不会包括它。我也看到了一些关于多边形的工作,但没有针对非规则形状的。
findMinRect
函数。该代码是Python原版此处的C++移植版本。
然后,您只需要为每种颜色创建一个掩码,在该掩码中查找每个斑点的连通组件,并计算每个斑点的最小矩形。
初始图像:
这里展示了每个色块找到的矩形,按颜色划分。然后您可以只选择所需的矩形,即每种颜色的最大矩形或最大色块的矩形。#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;
}