我建议你为你想要的每个形状创建一个包含颜色的标量向量。total_shape对应于您要着色的形状的一侧。例如,如果您想为包括八边形的形状着色,则total_shape = 8 + 1。将颜色存储到向量shape_colors。
std::vector<Scalar> shape_colors;
for (int i = 0; i < total_shape; i++)
{
Scalar color(rng.uniform(0, 255), rng.uniform(0, 255), rng.uniform(0, 255));
shape_colors.push_back(color);
}
然后,当您想为轮廓着色时,请检查每个轮廓有多少点。根据点编号,从 shape_color 中选择颜色,瞧,相同形状的相同配色方案。
但是,根据形状的角度,轮廓结果可能会返回太多点。我们需要使用 将轮廓简化为最简单的形式approxPolyDP
。这意味着我们希望一个矩形只包含 4 个点,一个三角形 3 个和一个五边形 5 个点。这个函数的详细解释由这个链接给出。通过这样做,我们将能够通过它包含的点的总数来确定轮廓的形状。
for (int i = 0; i < contours.size(); i++)
{
cv::Mat approx;
approxPolyDP(contours[i], approx, 30, true);
int n = approx.checkVector(2);
drawContours(dst, contours, i, shape_colors[n],25);
}
这是整个代码:
void process()
{
cv::Mat src;
cv::Mat dst;
cv::RNG rng;
std::string image_path = "Picture1.png";
src = cv::imread(image_path,0);
dst = cv::imread(image_path);
vector<vector<Point> > contours;
vector<Vec4i> hierarchy;
cv::threshold(src, src, 120, 255, cv::THRESH_BINARY);
findContours(src, contours, hierarchy,
CV_RETR_CCOMP, CV_CHAIN_APPROX_SIMPLE);
int total_shape = 10;
std::vector<Scalar> shape_colors;
for (int i = 0; i < total_shape; i++)
{
Scalar color(rng.uniform(0, 255), rng.uniform(0, 255), rng.uniform(0, 255));
shape_colors.push_back(color);
}
for (int i = 0; i < contours.size(); i++)
{
cv::Mat approx;
approxPolyDP(contours[i], approx, 30, true);
int n = approx.checkVector(2);
drawContours(dst, contours, i, shape_colors[n],25);
}
cv::imshow("dst", dst);
cv::waitKey(0);
}
这是结果: