opencv c++ 특정 객체의 위치를 구하는 방법

 

 

위의 이미지처럼 0과 255로 이원화된 이미지 속에서 흰색 박스의 가로 길이, 세로길이를 구하는 방법을 설명하고자 한다. 

 

x가 시작되는 부분과 x가 끝나는 부분을 구해서 서로 빼면 총 가로길이를 구할 수 있다. 

세로길이는 이와 똑같이 구할 수 있다.

 

이미지에 적용하여 구하려면, 흰색이 시작하는 부분과 흰색이 끝나는 부분을 구하면 길이를 구할 수 있다. 

 

 

이 부분을 코딩으로 구현한 부분은 아래와 같다. 

 

#pragma once
#include "IPDef.h"
#include "ISP.h"
 
int main(void)
{
    Mat img_color = imread("Square.bmp");
    int width = img_color.cols;
    int height = img_color.rows;
    Mat img_gray = Mat(img_color.rows, img_color.cols, CV_8UC1);
    ISP isp;
    isp.cvtColor(img_color.data, img_gray.data,
        img_color.cols, img_color.rows,
        ECvtColor::eCvtBGR2GRAY);
 
    int start_x = 0;
    int finish_x = 0;
    int start_y = 0;
    int finish_y = 0;
    bool bfind_x = false;
    bool bfind_y = false;
    uchar* pSrc = img_gray.data;
 
    //search start_y : half of width
    for (int row = 0; row < height - 1; row++)
    {
        int col = width / 2 - 1;
        int index_cur = row * width + col;
        int index_down = (row + 1) * width + col;
        int cur = pSrc[index_cur];
        int next = pSrc[index_down];
        if ((next - cur) == 255)
        {
            bfind_y = true;
            start_y = row;
            cout << "find start y = " << start_y << endl;
            break;
        }
    }
    //search finish_y : half of width
    for (int row = 0; row < height - 1; row++)
    {
        int col = width / 2 - 1;
        int index_cur = row * width + col;
        int index_down = (row + 1) * width + col;
        int cur = pSrc[index_cur];
        int next = pSrc[index_down];
        if ((next - cur) == -255)
        {
            finish_y = row;
            cout << "find finish y = " << finish_y << endl;
            cout << "complete y information" << endl;
            break;
        }
    }
 
    cout << "size_y = " << finish_y - start_y << endl;
 
 
    for (int col = 0; col < width - 1; col++)
    {
        int row = height / 2 - 1;
        int index_cur = row * width + col;
        int index_down = row * width + (col+1);
        int cur = pSrc[index_cur];
        int next = pSrc[index_down];
        if ((next - cur) == 255)
        {
            bfind_x = true;
            start_x = col;
            cout << "find start x = " << start_x << endl;
            break;
        }
    }
  
    for (int col = 0; col < width - 1; col++)
    {
        int row = height / 2 - 1;
        int index_cur = row * width + col;
        int index_down = row * width + (col+1);
        int cur = pSrc[index_cur];
        int next = pSrc[index_down];
        if ((next - cur) == -255)
        {
            finish_x = col;
            cout << "find finish x = " << finish_x << endl;
            cout << "complete x information" << endl;
            break;
        }
    }
 
    cout << "size_x = " << finish_x - start_x << endl;
 
 
    cin.get();
 
}