OpenCV——二值图赋色

一、概述

实现如下图所示的结果：

二、代码实现

#include<iostream>
#include <opencv2/opencv.hpp>

using namespace cv;
using namespace std;
int main()
{
	Mat image, src, binImage;
	image = cv::imread("二值图.png", -1);

	//cvtColor(image, src, COLOR_BGR2GRAY);
	//threshold(src, binImage, 1, 255, THRESH_BINARY);
	//
	Vec3b color = Vec3b(0, 0, 255);
	Mat preImage = Mat::zeros(image.size(), CV_8UC3);
	int w = preImage.cols;
	int h = preImage.rows;
	for (int row = 0; row < h; row++)
	{
		uchar* dataCurr = image.ptr<uchar>(row);
		for (int col = 0; col < w; col++)
		{
			int label = dataCurr[col];

			if (label != 0)
			{
				preImage.at<Vec3b>(row, col) = color;
			}
		}
	}
	imwrite("彩色.png",preImage);
	imshow("frame", preImage);
	waitKey(0);
}