我想在OpenCV(C (中从ATL::CImage
转换为cv::Mat
。您能帮您转换这个对象吗?
我从Windows屏幕截图(使用MFC(获得了CImage
。然后,我想在OpenCV Mat Object中处理图像。
我不知道如何转换。
- C 项目(VC 2017(
- MFC
- OPENCV 3.4.6
CImage image;
int cx;
int cy;
CWnd* pWndDesktop = CWnd::GetDesktopWindow();
CWindowDC srcDC(pWndDesktop);
Rect rcDesktopWindow;
::GetWindowRect(pWndDesktop->m_hWnd, %rcDesktopWindow);
cx = (rcDesktopWindow.right - rcDesktopWindow.left);
cy = (rcDesktopWindow.bottom - rcDesktopWindow.top);
image.create(cx, cy, srcDC.GetDeviceCaps(BITPIXEL));
CDC* pDC = CDC::FromHandle(image.GetDC());
pDC->BitBlt(0, 0, cx, cy, &srcDC, 0, 0, SRCCOPY);
image.ReleaseDC();
cv::Mat mat;
// I want set image to mat!
mat = image???
无法将ATL::Image
转换为cv::Mat
。
CImage
如果高度为正,则会创建一个底顶图。您必须通过负高度才能为mat
使用 CImage::GetBits
以下以下位置检索位:
HDC hdc = GetDC(0);
RECT rc;
GetClientRect(GetDesktopWindow(), &rc);
int cx = rc.right;
int cy = rc.bottom;
CImage image;
image.Create(cx, -cy, 32);
BitBlt(image.GetDC(), 0, 0, cx, cy, hdc, 0, 0, SRCCOPY);
image.ReleaseDC();
ReleaseDC(0, hdc);
cv::Mat mat;
mat.create(cy, cx, CV_8UC4);
memcpy(mat.data, image.GetBits(), cy * cx * 4);
//or borrow pixel data from CImage
cv::Mat mat(cy, cx, CV_8UC4, image.GetBits());
或强制深副本如下:
cv::Mat mat;
mat = cv::Mat(cy, cx, CV_8UC4, image.GetBits()).clone();
注意,CImage
为像素数据提供了自己的分配。Mat
需要进行自己的分配,或者必须从CImage
借用,这可能很棘手。
如果您只是进行屏幕截图,则可以使用普通的Windows API进行操作,然后直接写入cv::Mat
。这样,单个分配(更快一点(,而mat
不依赖其他对象。示例:
void foo()
{
HDC hdc = ::GetDC(0);
RECT rc;
::GetClientRect(::GetDesktopWindow(), &rc);
int cx = rc.right;
int cy = rc.bottom;
cv::Mat mat;
mat.create(cy, cx, CV_8UC4);
HBITMAP hbitmap = CreateCompatibleBitmap(hdc, cx, cy);
HDC memdc = CreateCompatibleDC(hdc);
HBITMAP oldbmp = (HBITMAP)SelectObject(memdc, hbitmap);
BitBlt(memdc, 0, 0, cx, cy, hdc, 0, 0, SRCCOPY);
BITMAPINFOHEADER bi = { sizeof(bi), cx, -cy, 1, 32, BI_RGB };
GetDIBits(hdc, hbitmap, 0, cy, mat.data, (BITMAPINFO*)&bi, DIB_RGB_COLORS);
//GDI cleanup:
SelectObject(memdc, oldbmp);
DeleteDC(memdc);
DeleteObject(hbitmap);
::ReleaseDC(0, hdc);
}
编辑:
将
mat.data = (unsigned char*)image.GetBits();
更改为memcpy(mat.data, image.GetBits(), cy * cx * 4);
将ReleaseDC(0, hdc)
更改为::ReleaseDC(0, hdc)
,以避免与CWnd::ReleaseDC(dc)
#include <opencv2opencv.hpp>
#include <opencv2/imgproc/types_c.h>
#include <atlimage.h>
using namespace cv;
Mat CImage2Mat(CImage cimg)
{
BITMAP bmp;
::GetObject(cimg.Detach(), sizeof(BITMAP), &bmp);
int nChannels = bmp.bmBitsPixel == 1 ? 1 : bmp.bmBitsPixel / 8;
int depth = bmp.bmBitsPixel == 1 ? IPL_DEPTH_1U : IPL_DEPTH_8U;
IplImage* img = cvCreateImageHeader(cvSize(bmp.bmWidth, bmp.bmHeight), depth, nChannels);
img->imageData = (char*)malloc(bmp.bmHeight * bmp.bmWidth * nChannels * sizeof(char));
memcpy(img->imageData, (char*)(bmp.bmBits), bmp.bmHeight * bmp.bmWidth * nChannels);
cvFlip(img, img, 0);
return cvarrToMat(static_cast<IplImage*>(img));
}
用法:
CImage cimg;
cimg.Load(L"resim.png");
Mat img = CImage2Mat(cimg);
通常,在MFC Windows应用程序开发中使用图像处理时,会发生这种情况。我正在开发一个基于MFC的应用程序,该应用使用EDSDK(Canon SDK(捕获佳能DSLR的图像。我使用了这个解决方案(节省时间(,我遇到的问题是图像分辨率的退化!
解决方案在这里!100%工作。https://github.com/wlrl/converttype