做毕设时遇到了一个问题,不能在MFC中显示由GrabCut抠出来的图。一番折腾后发现:
在OpenCV中如果图像最初是Mat类型的话,就不能通过转换为IplImage再转换为CvvImage来显示到View里(OpenCV里没有函数能直接将Mat显示到View里),只能通过StretchDIBits来绘制。
下面介绍StretchDIBits函数的使用方法:
1)创建MFC单文档项目(我创建了UseOpenCV),设置包含目录与库目录,添加相关附加依赖项。
2)在文档头文件(UseOpenCVDoc.h)中添加包含文件:
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
添加OpenCV Mat类类型成员变量 m_cvImg,形式如下:
protected: // 仅从序列化创建
CUseOpenCVDoc();
DECLARE_DYNCREATE(CUseOpenCVDoc)
cv::Mat m_cvImg;
3)在文档实现文件(UseOpenCVDoc.cpp)中重写OnOpenDocument函数,如下:
BOOL CUseOpenCVDoc::OnOpenDocument(LPCTSTR lpszPathName)
{
if (!CDocument::OnOpenDocument(lpszPathName))
return FALSE;
if (lpszPathName != NULL)
{
// 将lpszPathName 转换成 std::string类型 文件名
#ifdef UNICODE
// 中文路径时会存在问题
size_t i = 0;
char* nstring = new char[(wcslen(lpszPathName)+1)*2];
wcstombs_s(&i, nstring, (wcslen(lpszPathName)+1)*2, lpszPathName, (wcslen(lpszPathName)+1)*2);
std::string filename(nstring);
delete[] nstring;
#else
std::string filename(lpszPathName);
#endif
m_cvImg = cv::imread(filename);
if (!m_cvImg.data)
{
MessageBox(NULL, lpszPathName, _T("图像文件打开失败"), MB_OK);
return FALSE;
}
}
return TRUE;
}
4)在视图头文件(UseOpenCVView.h)中添加四个成员变量,如下:
protected:
cv::Mat m_cvImg;
BITMAPINFO* m_bmi;
BITMAPINFOHEADER* m_bmih;
unsigned int m_buffer[sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD)*256];
5)修改视图类 (CUseOpenCVView)构造函数如下:
CUseOpenCVView::CUseOpenCVView()
{
// TODO: 在此处添加构造代码
// 初始化 BITMAPINFO结构 及 BITMAPINFOHEADER结构
m_bmi = (BITMAPINFO*) m_buffer;
m_bmih = &(m_bmi->bmiHeader);
memset(m_bmih, 0, sizeof(*m_bmih));
m_bmih->biSize = sizeof(BITMAPINFOHEADER);
}
6)修改视图绘制函数如下:
void CUseOpenCVView::OnDraw(CDC* pDC)
{
CUseOpenCVDoc* pDoc = GetDocument();
ASSERT_VALID(pDoc);
if (!pDoc)
return;
// TODO: 在此处为本机数据添加绘制代码
m_cvImg = pDoc->m_cvImg;
m_bmih->biWidth = m_cvImg.cols;
m_bmih->biHeight = -m_cvImg.rows; // 在自下而上的位图中 高度为负
m_bmih->biPlanes = 1;
m_bmih->biCompression = BI_RGB;
m_bmih->biBitCount = 8 * m_cvImg.channels();
CRect rect;
GetClientRect(&rect);
StretchDIBits(
pDC->GetSafeHdc(),
0, 0, rect.Width(), rect.Height(),
0, 0, m_cvImg.cols, m_cvImg.rows,
m_cvImg.data,
(BITMAPINFO*) m_bmi,
DIB_RGB_COLORS,
SRCCOPY);
}
编译运行程序,打开OpenCV支持的图像文件后,图像就绘制在视图中了。