下面是一个使用C++和OpenCV自定义colormap的示例代码:
#include <opencv2/opencv.hpp>
// 自定义colormap函数
void customColorMap(const cv::Mat& src, cv::Mat& dst)
{
CV_Assert(src.type() == CV_8UC1); // 确保输入图像是8位单通道灰度图像
// 创建一个空的输出图像
dst.create(src.size(), CV_8UC3);
for (int y = 0; y < src.rows; ++y)
{
for (int x = 0; x < src.cols; ++x)
{
uchar grayValue = src.at<uchar>(y, x);
// 自定义映射规则,这里将灰度值映射到蓝色-红色渐变色上
uchar blue = 255 - grayValue;
uchar green = 0;
uchar red = grayValue;
dst.at<cv::Vec3b>(y, x) = cv::Vec3b(blue, green, red);
}
}
}
int main()
{
cv::Mat grayImage = cv::imread("gray_image.png", cv::IMREAD_GRAYSCALE);
if (grayImage.empty())
{
std::cout << "Failed to load image." << std::endl;
return -1;
}
cv::Mat colorImage;
// 应用自定义colormap函数生成彩色图像
customColorMap(grayImage, colorImage);
cv::imshow("Custom Color Image", colorImage);
cv::waitKey(0);
return 0;
}
在此示例中,我们定义了一个名为customColorMap
的自定义colormap函数。它接受一个灰度图像作为输入,并生成相应的彩色图像。
在自定义映射规则中,我们将灰度值映射到蓝色-红色渐变色上。通过遍历输入图像的每个像素,并根据灰度值计算对应的蓝、绿、红分量,然后将这些分量赋给输出图像的相应通道。
你可以根据需要修改自定义colormap函数,实现你所需的颜色映射效果。
内容由零声教学AI助手提供,问题来源于学员提问