gpt4 book ai didi

c++ - 将 FreeImage FIBITMAP 格式转换为 OpenCV Mat

转载 作者:太空宇宙 更新时间:2023-11-04 15:36:38 25 4
gpt4 key购买 nike

我一直在寻找答案,但我能找到的只是与我需要的相反的答案(Mat 到 FIBITMAP)。我有一个简短的代码,可以将图像加载为 FIBITMAP 对象。我需要 FreeImage,因为 OpenCV 不允许我加载 pfm 和其他一些扩展文件。我想要做的是将 FIBTMAP 图像转换为 OpenCV Mat,这样我就可以在我已有的更长的代码中使用它。我该怎么做?

最佳答案

FI2MAT 函数将执行从 FreeImage 到 OpenCV Mat 的转换。

我现在无法彻底测试每种类型的 FreeImage 格式,但它应该可以按预期工作。让我知道某些格式是否存在问题。

请记住,imshow 可能无法正确呈现所有类型,因此您可能需要先转换图像。

这里有一个例子的代码:

#include <FreeImage.h>
#include <opencv2\opencv.hpp>
using namespace cv;

void FI2MAT(FIBITMAP* src, Mat& dst)
{
//FIT_BITMAP //standard image : 1 - , 4 - , 8 - , 16 - , 24 - , 32 - bit
//FIT_UINT16 //array of unsigned short : unsigned 16 - bit
//FIT_INT16 //array of short : signed 16 - bit
//FIT_UINT32 //array of unsigned long : unsigned 32 - bit
//FIT_INT32 //array of long : signed 32 - bit
//FIT_FLOAT //array of float : 32 - bit IEEE floating point
//FIT_DOUBLE //array of double : 64 - bit IEEE floating point
//FIT_COMPLEX //array of FICOMPLEX : 2 x 64 - bit IEEE floating point
//FIT_RGB16 //48 - bit RGB image : 3 x 16 - bit
//FIT_RGBA16 //64 - bit RGBA image : 4 x 16 - bit
//FIT_RGBF //96 - bit RGB float image : 3 x 32 - bit IEEE floating point
//FIT_RGBAF //128 - bit RGBA float image : 4 x 32 - bit IEEE floating point

int bpp = FreeImage_GetBPP(src);
FREE_IMAGE_TYPE fit = FreeImage_GetImageType(src);

int cv_type = -1;
int cv_cvt = -1;

switch (fit)
{
case FIT_UINT16: cv_type = DataType<ushort>::type; break;
case FIT_INT16: cv_type = DataType<short>::type; break;
case FIT_UINT32: cv_type = DataType<unsigned>::type; break;
case FIT_INT32: cv_type = DataType<int>::type; break;
case FIT_FLOAT: cv_type = DataType<float>::type; break;
case FIT_DOUBLE: cv_type = DataType<double>::type; break;
case FIT_COMPLEX: cv_type = DataType<Complex<double>>::type; break;
case FIT_RGB16: cv_type = DataType<Vec<ushort, 3>>::type; cv_cvt = COLOR_RGB2BGR; break;
case FIT_RGBA16: cv_type = DataType<Vec<ushort, 4>>::type; cv_cvt = COLOR_RGBA2BGRA; break;
case FIT_RGBF: cv_type = DataType<Vec<float, 3>>::type; cv_cvt = COLOR_RGB2BGR; break;
case FIT_RGBAF: cv_type = DataType<Vec<float, 4>>::type; cv_cvt = COLOR_RGBA2BGRA; break;
case FIT_BITMAP:
switch (bpp) {
case 8: cv_type = DataType<Vec<uchar, 1>>::type; break;
case 16: cv_type = DataType<Vec<uchar, 2>>::type; break;
case 24: cv_type = DataType<Vec<uchar, 3>>::type; break;
case 32: cv_type = DataType<Vec<uchar, 4>>::type; break;
default:
// 1, 4 // Unsupported natively
cv_type = -1;
}
break;
default:
// FIT_UNKNOWN // unknown type
dst = Mat(); // return empty Mat
return;
}

int width = FreeImage_GetWidth(src);
int height = FreeImage_GetHeight(src);
int step = FreeImage_GetPitch(src);

if (cv_type >= 0) {
dst = Mat(height, width, cv_type, FreeImage_GetBits(src), step);
if (cv_cvt > 0)
{
cvtColor(dst, dst, cv_cvt);
}
}
else {

vector<uchar> lut;
int n = pow(2, bpp);
for (int i = 0; i < n; ++i)
{
lut.push_back(static_cast<uchar>((255 / (n - 1))*i));
}

FIBITMAP* palletized = FreeImage_ConvertTo8Bits(src);
BYTE* data = FreeImage_GetBits(src);
for (int r = 0; r < height; ++r) {
for (int c = 0; c < width; ++c) {
dst.at<uchar>(r, c) = saturate_cast<uchar>(lut[data[r*step + c]]);
}
}
}

flip(dst, dst, 0);
}

int main()
{
FreeImage_Initialise();
FREE_IMAGE_FORMAT format = FreeImage_GetFileType("path_to_image", 0);
FIBITMAP* fi_image = FreeImage_Load(format, "path_to_image");

Mat cv_img;
FI2MAT(fi_image, cv_img);

return 0;
}

关于c++ - 将 FreeImage FIBITMAP 格式转换为 OpenCV Mat,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31814451/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com