Realsense 提取彩色和深度視頻流


一、簡要介紹

        關於realsense的介紹,網上很多,這里不再贅述,sdk及相關文檔可參考realsense SDK,也可參考開發人員專區

運行代碼之前,要確保你已經安裝好了realsense的DCM和SDK,官網有教程,具體請參考DCM和SDK安裝步驟

二、代碼

 

  1. #include <pxcsensemanager.h>    
  2. #include <pxcsession.h>    
  3. #include "util_render.h"    
  4. #include <iostream>    
  5. #include <string>    
  6. #include <stdio.h>    
  7. #include <opencv2\opencv.hpp>    
  8. #include <windows.h>  
  9.   
  10. #define WIDTH 640    
  11. #define HEIGHT 480    
  12.   
  13. using namespace cv;  
  14. using namespace std;  
  15.   
  16. int main(int argc, char** argv)  
  17. {  
  18.     UtilRender *renderColor = new UtilRender(L"COLOR_STREAM");  
  19.     UtilRender *renderDepth = new UtilRender(L"DEPTH_STREAM");  
  20.   
  21.     PXCSenseManager *psm = 0;  
  22.     psm = PXCSenseManager::CreateInstance();  
  23.     if (!psm)  
  24.     {  
  25.         wprintf_s(L"Unabel to create the PXCSenseManager\n");  
  26.         return 1;  
  27.     }  
  28.     pxcStatus sts;  
  29.   
  30.     psm->EnableStream(PXCCapture::STREAM_TYPE_COLOR, WIDTH, HEIGHT);  
  31.       
  32.     psm->EnableStream(PXCCapture::STREAM_TYPE_DEPTH, WIDTH, HEIGHT);  
  33.       
  34.     sts = psm->Init();  
  35.     if (sts != PXC_STATUS_NO_ERROR)  
  36.     {  
  37.         wprintf_s(L"Unabel to Initializes the pipeline\n");  
  38.         return 2;  
  39.     }  
  40.       
  41.     PXCImage *colorIm, *depthIm;  
  42.     PXCImage::ImageData depth_data,color_data;  
  43.     PXCImage::ImageInfo depth_info,color_info;  
  44.     while (psm->AcquireFrame(true) >= PXC_STATUS_NO_ERROR)  
  45.   
  46.     {  
  47.         if (psm->AcquireFrame(true) < PXC_STATUS_NO_ERROR) break;  
  48.   
  49.   
  50.         PXCCapture::Sample *sample = psm->QuerySample();  
  51.   
  52.         colorIm = sample->color;  
  53.         depthIm = sample->depth;  
  54.   
  55.         if (colorIm->AcquireAccess(PXCImage::ACCESS_READ, PXCImage::PIXEL_FORMAT_RGB24, &color_data) < PXC_STATUS_NO_ERROR)  
  56.             wprintf_s(L"未正常獲取彩色圖\n");  
  57.         if (depthIm->AcquireAccess(PXCImage::ACCESS_READ, &depth_data) < PXC_STATUS_NO_ERROR)  
  58.             wprintf_s(L"未正常獲取深度圖\n");  
  59.           
  60.         depth_info = sample->depth->QueryInfo();  
  61.         color_info = sample->color->QueryInfo();  
  62.   
  63.         Mat depth(Size(depth_info.width, depth_info.height), CV_16UC1, (void*)depth_data.planes[0], depth_data.pitches[0] / sizeof(uchar));  
  64.         Mat color(Size(color_info.width, color_info.height), CV_8UC3, (void*)color_data.planes[0], color_data.pitches[0] / sizeof(uchar));  
  65.   
  66.         depthIm->ReleaseAccess(&depth_data);  
  67.         colorIm->ReleaseAccess(&color_data);  
  68.   
  69.         if (!renderColor->RenderFrame(colorIm)) break;  
  70.         if (!renderDepth->RenderFrame(depthIm)) break;  
  71.   
  72.         psm->ReleaseFrame();  
  73.   
  74.         imshow("color", color);  
  75.         waitKey(1);  
  76.         //CV_16UC1的圖片在imshow時會除以256,將最遠探測距離設為z,那么imshow時可以乘以255*256/z,此處乘以15  
  77.         imshow("depth", depth * 15);  
  78.         waitKey(1);  
  79.   
  80.     }  
  81.     psm->Release();  
  82.     system("pause");  
  83. }  
#include <pxcsensemanager.h>  
#include <pxcsession.h>  
#include "util_render.h"  
#include <iostream>  
#include <string>  
#include <stdio.h>  
#include <opencv2\opencv.hpp>  
#include <windows.h>

#define WIDTH 640  
#define HEIGHT 480  

using namespace cv;
using namespace std;

int main(int argc, char** argv)
{
	UtilRender *renderColor = new UtilRender(L"COLOR_STREAM");
	UtilRender *renderDepth = new UtilRender(L"DEPTH_STREAM");

	PXCSenseManager *psm = 0;
	psm = PXCSenseManager::CreateInstance();
	if (!psm)
	{
		wprintf_s(L"Unabel to create the PXCSenseManager\n");
		return 1;
	}
	pxcStatus sts;

	psm->EnableStream(PXCCapture::STREAM_TYPE_COLOR, WIDTH, HEIGHT);
	
	psm->EnableStream(PXCCapture::STREAM_TYPE_DEPTH, WIDTH, HEIGHT);
	
	sts = psm->Init();
	if (sts != PXC_STATUS_NO_ERROR)
	{
		wprintf_s(L"Unabel to Initializes the pipeline\n");
		return 2;
	}
	
	PXCImage *colorIm, *depthIm;
	PXCImage::ImageData depth_data,color_data;
	PXCImage::ImageInfo depth_info,color_info;
	while (psm->AcquireFrame(true) >= PXC_STATUS_NO_ERROR)

	{
		if (psm->AcquireFrame(true) < PXC_STATUS_NO_ERROR) break;


		PXCCapture::Sample *sample = psm->QuerySample();

		colorIm = sample->color;
		depthIm = sample->depth;

		if (colorIm->AcquireAccess(PXCImage::ACCESS_READ, PXCImage::PIXEL_FORMAT_RGB24, &color_data) < PXC_STATUS_NO_ERROR)
			wprintf_s(L"未正常獲取彩色圖\n");
		if (depthIm->AcquireAccess(PXCImage::ACCESS_READ, &depth_data) < PXC_STATUS_NO_ERROR)
			wprintf_s(L"未正常獲取深度圖\n");
		
		depth_info = sample->depth->QueryInfo();
		color_info = sample->color->QueryInfo();

		Mat depth(Size(depth_info.width, depth_info.height), CV_16UC1, (void*)depth_data.planes[0], depth_data.pitches[0] / sizeof(uchar));
		Mat color(Size(color_info.width, color_info.height), CV_8UC3, (void*)color_data.planes[0], color_data.pitches[0] / sizeof(uchar));

		depthIm->ReleaseAccess(&depth_data);
		colorIm->ReleaseAccess(&color_data);

		if (!renderColor->RenderFrame(colorIm)) break;
		if (!renderDepth->RenderFrame(depthIm)) break;

		psm->ReleaseFrame();

		imshow("color", color);
		waitKey(1);
		//CV_16UC1的圖片在imshow時會除以256,將最遠探測距離設為z,那么imshow時可以乘以255*256/z,此處乘以15
		imshow("depth", depth * 15);
		waitKey(1);

	}
	psm->Release();
	system("pause");
}

三、相關解釋

 

        新建項目,寫入源代碼,生成解決方案之前,記得要在項目的屬性管理器右鍵項目,選擇添加現有屬性表,選擇C:\Program Files (x86)\Intel\RSSDK\props目錄下的一個官方配置好的屬性表(前提是DCM和SDK的安裝路徑是默認路徑,沒有自行修改);另外因為我們還要進行cv::Mat類型的轉化,所以也要導入opencv的屬性表。

        運行之后,可以看到4個窗口,COLOR_STREAM和DEPTH_STREAM為分別realsense提取到的彩色視頻流和深度視頻流;color和depth分別為我們轉化為opencv中mat類型之后的彩色和深度視頻流。

目前實現pxcimage到mat類型轉化的方法有好幾種,大體類似,主要是對PXCImage::ImageData中的plane和pitch的理解,博主理解為plane[0]為該sdk中圖片數據的首地址,pitches可參考http://www.cnblogs.com/gamedes/p/4541765.html

也看到論壇上有人專門寫過pxcimage到mat類型的轉化函數,函數代碼如下:

 

  1. void ConvertPXCImageToOpenCVMat(PXCImage *inImg, Mat *outImg) {  
  2.     int cvDataType;  
  3.     int cvDataWidth;  
  4.   
  5.   
  6.     PXCImage::ImageData data;  
  7.     inImg->AcquireAccess(PXCImage::ACCESS_READ, &data);  
  8.     PXCImage::ImageInfo imgInfo = inImg->QueryInfo();  
  9.   
  10.     switch (data.format) {  
  11.         /* STREAM_TYPE_COLOR */  
  12.         case PXCImage::PIXEL_FORMAT_YUY2: /* YUY2 image  */  
  13.         case PXCImage::PIXEL_FORMAT_NV12: /* NV12 image */  
  14.             throw(0); // Not implemented  
  15.         case PXCImage::PIXEL_FORMAT_RGB32: /* BGRA layout on a little-endian machine */  
  16.             cvDataType = CV_8UC4;  
  17.             cvDataWidth = 4;  
  18.             break;  
  19.         case PXCImage::PIXEL_FORMAT_RGB24: /* BGR layout on a little-endian machine */  
  20.             cvDataType = CV_8UC3;  
  21.             cvDataWidth = 3;  
  22.             break;  
  23.         case PXCImage::PIXEL_FORMAT_Y8:  /* 8-Bit Gray Image, or IR 8-bit */  
  24.             cvDataType = CV_8U;  
  25.             cvDataWidth = 1;  
  26.             break;  
  27.   
  28.         /* STREAM_TYPE_DEPTH */  
  29.         case PXCImage::PIXEL_FORMAT_DEPTH: /* 16-bit unsigned integer with precision mm. */  
  30.         case PXCImage::PIXEL_FORMAT_DEPTH_RAW: /* 16-bit unsigned integer with device specific precision (call device->QueryDepthUnit()) */  
  31.             cvDataType = CV_16U;  
  32.             cvDataWidth = 2;  
  33.             break;  
  34.         case PXCImage::PIXEL_FORMAT_DEPTH_F32: /* 32-bit float-point with precision mm. */  
  35.             cvDataType = CV_32F;  
  36.             cvDataWidth = 4;  
  37.             break;  
  38.   
  39.         /* STREAM_TYPE_IR */  
  40.         case PXCImage::PIXEL_FORMAT_Y16:          /* 16-Bit Gray Image */  
  41.             cvDataType = CV_16U;  
  42.             cvDataWidth = 2;  
  43.             break;  
  44.         case PXCImage::PIXEL_FORMAT_Y8_IR_RELATIVE:    /* Relative IR Image */  
  45.             cvDataType = CV_8U;  
  46.             cvDataWidth = 1;  
  47.             break;  
  48.         }  
  49.   
  50.     // suppose that no other planes  
  51.     if (data.planes[1] != NULL) throw(0); // not implemented  
  52.     // suppose that no sub pixel padding needed  
  53.     if (data.pitches[0] % cvDataWidth!=0) throw(0); // not implemented  
  54.   
  55.     outImg->create(imgInfo.height, data.pitches[0] / cvDataWidth, cvDataType);  
  56.   
  57.     memcpy(outImg->data, data.planes[0], imgInfo.height*imgInfo.width*cvDataWidth*sizeof(pxcBYTE));  
  58.   
  59.     inImg->ReleaseAccess(&data);  
  60.     }  
void ConvertPXCImageToOpenCVMat(PXCImage *inImg, Mat *outImg) {
    int cvDataType;
    int cvDataWidth;


    PXCImage::ImageData data;
    inImg->AcquireAccess(PXCImage::ACCESS_READ, &data);
    PXCImage::ImageInfo imgInfo = inImg->QueryInfo();

    switch (data.format) {
        /* STREAM_TYPE_COLOR */
        case PXCImage::PIXEL_FORMAT_YUY2: /* YUY2 image  */
        case PXCImage::PIXEL_FORMAT_NV12: /* NV12 image */
            throw(0); // Not implemented
        case PXCImage::PIXEL_FORMAT_RGB32: /* BGRA layout on a little-endian machine */
            cvDataType = CV_8UC4;
            cvDataWidth = 4;
            break;
        case PXCImage::PIXEL_FORMAT_RGB24: /* BGR layout on a little-endian machine */
            cvDataType = CV_8UC3;
            cvDataWidth = 3;
            break;
        case PXCImage::PIXEL_FORMAT_Y8:  /* 8-Bit Gray Image, or IR 8-bit */
            cvDataType = CV_8U;
            cvDataWidth = 1;
            break;

        /* STREAM_TYPE_DEPTH */
        case PXCImage::PIXEL_FORMAT_DEPTH: /* 16-bit unsigned integer with precision mm. */
        case PXCImage::PIXEL_FORMAT_DEPTH_RAW: /* 16-bit unsigned integer with device specific precision (call device->QueryDepthUnit()) */
            cvDataType = CV_16U;
            cvDataWidth = 2;
            break;
        case PXCImage::PIXEL_FORMAT_DEPTH_F32: /* 32-bit float-point with precision mm. */
            cvDataType = CV_32F;
            cvDataWidth = 4;
            break;

        /* STREAM_TYPE_IR */
        case PXCImage::PIXEL_FORMAT_Y16:          /* 16-Bit Gray Image */
            cvDataType = CV_16U;
            cvDataWidth = 2;
            break;
        case PXCImage::PIXEL_FORMAT_Y8_IR_RELATIVE:    /* Relative IR Image */
            cvDataType = CV_8U;
            cvDataWidth = 1;
            break;
        }

    // suppose that no other planes
    if (data.planes[1] != NULL) throw(0); // not implemented
    // suppose that no sub pixel padding needed
    if (data.pitches[0] % cvDataWidth!=0) throw(0); // not implemented

    outImg->create(imgInfo.height, data.pitches[0] / cvDataWidth, cvDataType);

    memcpy(outImg->data, data.planes[0], imgInfo.height*imgInfo.width*cvDataWidth*sizeof(pxcBYTE));

    inImg->ReleaseAccess(&data);
    }

原則上傳入pxcimage的指針,再通過case根據原圖像類型選擇要轉化的圖像類型是沒有問題的,博主嘗試過使用這個函數,但總會出現R1060的錯誤,似乎是指針非法使用。大家可以自行嘗試一下,成功的小伙伴麻煩告知一下博主,讓博主學習學習。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM