从像素标签创建RGB图像

时间:2016-03-14 17:12:13

标签: c++ opencv image-processing

给定CV_32SC1 cv::Mat图像包含每个像素的标签(标签只是0..N-1中的索引),OpenCV中生成CV_8UC3图像的最干净的代码是什么它显示每个连接的组件具有不同的任意颜色?如果我不必手动指定颜色,就像cv::floodFill一样,那就更好了。

1 个答案:

答案 0 :(得分:3)

如果标签的最大数量为256,您可以使用applyColorMap,将图片转换为Mat1i img = ... // Convert to CV_8U Mat1b img2; img.convertTo(img2, CV_8U); // Apply color map Mat3b out; applyColorMap(img2, out, COLORMAP_JET);

jet

如果标签数量高于256,则需要自己完成。下面是一个生成JET色图的示例(它基于//Create JET colormap函数的Matlab实现)。然后,您可以为矩阵的每个元素应用色彩映射。

请注意,如果您需要不同的色彩图或随机颜色,则只需修改#include <opencv2/opencv.hpp> #include <algorithm> using namespace std; using namespace cv; void applyCustomColormap(const Mat1i& src, Mat3b& dst) { // Create JET colormap double m; minMaxLoc(src, nullptr, &m); m++; int n = ceil(m / 4); Mat1d u(n*3-1, 1, double(1.0)); for (int i = 1; i <= n; ++i) { u(i-1) = double(i) / n; u((n*3-1) - i) = double(i) / n; } vector<double> g(n * 3 - 1, 1); vector<double> r(n * 3 - 1, 1); vector<double> b(n * 3 - 1, 1); for (int i = 0; i < g.size(); ++i) { g[i] = ceil(double(n) / 2) - (int(m)%4 == 1 ? 1 : 0) + i + 1; r[i] = g[i] + n; b[i] = g[i] - n; } g.erase(remove_if(g.begin(), g.end(), [m](double v){ return v > m;}), g.end()); r.erase(remove_if(r.begin(), r.end(), [m](double v){ return v > m; }), r.end()); b.erase(remove_if(b.begin(), b.end(), [](double v){ return v < 1.0; }), b.end()); Mat1d cmap(m, 3, double(0.0)); for (int i = 0; i < r.size(); ++i) { cmap(int(r[i])-1, 2) = u(i); } for (int i = 0; i < g.size(); ++i) { cmap(int(g[i])-1, 1) = u(i); } for (int i = 0; i < b.size(); ++i) { cmap(int(b[i])-1, 0) = u(u.rows - b.size() + i); } Mat3d cmap3 = cmap.reshape(3); Mat3b colormap; cmap3.convertTo(colormap, CV_8U, 255.0); // Apply color mapping dst = Mat3b(src.rows, src.cols, Vec3b(0,0,0)); for (int r = 0; r < src.rows; ++r) { for (int c = 0; c < src.cols; ++c) { dst(r, c) = colormap(src(r,c)); } } } int main() { Mat1i img(1000,1000); randu(img, Scalar(0), Scalar(10)); Mat3b out; applyCustomColormap(img, out); imshow("Result", out); waitKey(); return 0; } 部分:

$(document).on('click','your-element-class', function() {
    if ($(this).hasClass('active')) {
        $(this).removeClass('active');
    } else {
        $(this).addClass('active');
    }
});