OpenCV 흑백영상 추출하기

// ConsoleCV.cpp : Defines the entry point for the console application. // #include "stdafx.h" #include "20063703.h" // OpenCV 관련 헤더파일 #include <cv.h> #include <cxcore.h> #include <highgui.h> #include <cvcam.h> #ifdef _DEBUG #define new DEBUG_NEW #undef THIS_FILE static char THIS_FILE[] = __FILE__; #endif ///////////////////////////////////////////////////////////////////////////// // The one and only application object CWinApp theApp; using namespace std; void imageViewer(); char *getFileDialog(); void directAccessPixel(); void funcAccessPixel(); int _tmain(int argc, TCHAR* argv[], TCHAR* envp[]) { int nRetCode = 0; // initialize MFC and print and error on failure if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0)) { // TODO: change error code to suit your needs cerr << _T("Fatal Error: MFC initialization failed") << endl; nRetCode = 1; } else { // TODO: code your application's behavior here. //imageViewer( ); //directAccessPixel(); funcAccessPixel(); } return nRetCode; } void imageViewer( ) { IplImage *color_image, *gray_image; // STEP 1 : 영상 파일 선택 char *readFileName = getFileDialog(); if( !readFileName ) return; // STEP 2 : 컬러 영상을 읽는다. color_image = cvLoadImage( readFileName, CV_LOAD_IMAGE_COLOR ); // STEP 3 : 읽기 성공이면 if( color_image != 0 ) { // 캡션 제목이 title1 을 갖는 창을 생성 char *title1 = "First Image Viewer using OpenCV"; cvNamedWindow( title1, CV_WINDOW_AUTOSIZE ); // 생성된 창에 image 객체를 뿌려준다. cvShowImage( title1, color_image ); // 변환 결과를 담을 명암도 영상을 초기화 한다. gray_image = cvCreateImage( cvGetSize(color_image), IPL_DEPTH_8U, 1 ); // 컬러 영상을 명암도 영상으로 변환한다. cvCvtColor( color_image, gray_image, CV_BGR2GRAY ); // 캡션 제목이 title2 을 갖는 창을 생성 char *title2 = "First Gray Viewer using OpenCV"; cvNamedWindow( title2, CV_WINDOW_AUTOSIZE ); // 생성된 창에 image 객체를 뿌려준다. cvShowImage( title2, gray_image ); cvWaitKey(0); // 키보드 입력을 기다린다. cvReleaseImage( &color_image ); // 할당된 리소스를 해제한다. cvReleaseImage( &gray_image ); cvDestroyWindow( title1 ); cvDestroyWindow( title2 ); // 창을 닫는다. } else printf("[ERROR] 영상 파일을 읽어올 수 없습니다.\n"); return; } // 파일 대화상자에서 갖고 온다. char *getFileDialog() { char *readFileName = (char *)NULL; // 파일 포맷 char szFilter[] = "지원 영상 파일(*.bmp, *.jpg, *.gif, *.png, *.tif) \ |*.bmp;*.jpg;*.gif;*.png;*.tif||"; printf("\n"); printf("\t * 파일 대화상자를 이용하여 영상 파일을 선택하십시오.\n\n"); printf("\t \n"); CFileDialog fileDlg(TRUE, NULL, NULL, OFN_EXPLORER|OFN_HIDEREADONLY, szFilter); if( fileDlg.DoModal() == IDOK ) { // CString to char * readFileName = (char *)strdup((char *)(LPCTSTR)fileDlg.GetPathName()); } return readFileName; } /* void directAccessPixel() { int i,j; BYTE r,g,b,var; // BYTE는 UNSIGNED CHAR의 다른 표현이다 // 영상 데이터는 0~255까지의 값을 가지므로, 이에 대응하는 자료형인 BYTE로 선언하는 것 char *readFileName = getFileDialog(); if(!readFileName) return; IplImage *src_image = cvLoadImage(readFileName, CV_LOAD_IMAGE_COLOR); IplImage *dst_image = cvCreateImage(cvGetSize(src_image),IPL_DEPTH_8U,1); for(i=0;i<dst_image->height;i++) { for(j=0;j<dst_image->width;j++) { b=(BYTE)src_image->imageData[(i*src_image->widthStep) +j*3+0]; g=(BYTE)src_image->imageData[(i*src_image->widthStep) +j*3+1]; r=(BYTE)src_image->imageData[(i*src_image->widthStep) +j*3+2]; var=(BYTE)((r+g+b)/3.0); dst_image->imageData[(i*dst_image->widthStep)+j]=var; } } for(i=100;i<=300;i++) { for(j=50;j<=150;j++){ dst_image->imageData[(i*dst_image->widthStep)+j]=(BYTE)255; // char에서 int로 캐스팅 } } for(i=100;i<300;i++) { for(j=300;j<400;j++) if(j==300 || j==399 || i==100 || i==299) { dst_image->imageData[(i*dst_image->widthStep)+j]=(BYTE)255; } } for(i=100;i<=300;i++) { for(j=50;j<=150;j++) { src_image->imageData[(i*src_image->widthStep) +j*3+0]=(BYTE)255; src_image->imageData[(i*src_image->widthStep) +j*3+1]=(BYTE)255; src_image->imageData[(i*src_image->widthStep) +j*3+2]=(BYTE)255; } } for(i=100;i<300;i++) { for(j=300;j<400;j++) if(j==300 || j==399 || i==100 || i==299) { src_image->imageData[(i*src_image->widthStep) +j*3+0]=(BYTE)255; src_image->imageData[(i*src_image->widthStep) +j*3+1]=(BYTE)255; src_image->imageData[(i*src_image->widthStep) +j*3+2]=(BYTE)255; } } char *title = "명암도 영상"; cvNamedWindow(title, CV_WINDOW_AUTOSIZE); cvShowImage(title, dst_image); char *title2 = "흰색 명암도 영상"; cvNamedWindow(title2, CV_WINDOW_AUTOSIZE); cvShowImage(title2, src_image); cvWaitKey(0); cvReleaseImage(&src_image); cvReleaseImage(&dst_image); cvDestroyWindow(title); } */ void funcAccessPixel() { int i, j; double r, g, b, var; char *readFileName = getFileDialog(); if(!readFileName) return; IplImage *src_image = cvLoadImage( readFileName, CV_LOAD_IMAGE_COLOR ); // 분할한다. IplImage *src_red = cvCreateImage( cvGetSize(src_image), IPL_DEPTH_8U, 1); IplImage *src_green = cvCreateImage( cvGetSize(src_image), IPL_DEPTH_8U, 1); IplImage *src_blue = cvCreateImage( cvGetSize(src_image), IPL_DEPTH_8U, 1); // RGB 컬러공간을 Red, Green, Blue 채널로 분리한다. cvCvtPixToPlane( src_image, src_blue, src_green, src_red, NULL ); // 명암도 영상 IplImage *dst_image = cvCreateImage( cvGetSize(src_image), IPL_DEPTH_8U, 1); for(i=0; i<src_image->height; i++) { for(j=0; j<src_image->width; j++) { r = cvGetReal2D( src_red, i, j ); g = cvGetReal2D( src_green, i, j ); b = cvGetReal2D( src_blue, i, j ); var = (r + g + b)/3.0; cvSetReal2D( dst_image, i, j, var ); } } for(i=100;i<=300;i++) { for(j=50;j<=150;j++){ dst_image->imageData[(i*dst_image->widthStep)+j]=(BYTE)225; // char에서 int로 캐스팅 } } for(i=100;i<300;i++) { for(j=300;j<400;j++) if(j==300 || j==399 || i==100 || i==299) { dst_image->imageData[(i*dst_image->widthStep)+j]=(BYTE)225; } } //결과 출력 char *title = "명암도 영상"; cvNamedWindow(title, CV_WINDOW_AUTOSIZE); cvShowImage(title, dst_image); cvWaitKey(0); cvReleaseImage(&src_image); cvReleaseImage(&src_red); cvReleaseImage(&src_green); cvReleaseImage(&src_blue); cvReleaseImage(&src_image); cvDestroyWindow(title); }