複数の画像の形状を比較する OpenCV Trackback() Comment(0)

複数の画像（0.png、1.png、…）を読み込み、そのうちの一枚と他の画像の形状を比較するプログラムを練習用に作ってみた。練習用なのでエラーチェックが適当かもです。

#include <stdlib.h>
#include <iostream>
#include <sstream>
#include <vector>
#include <cv.h>
#include <cxcore.h>
#include <highgui.h>

int main(int argc, char* argv[])
{
    int i;

    // 画像の読み込み枚数を指定
    std::cout << "Enter the number of image.\n";
    std::cin >> i;

    const int num(i);

    std::vector<IplImage*> img(num);

    {
        // 画像読み込み
        std::stringstream ssm;
        std::string fname;
        for(i = 0; i < num; ++i){
            ssm << "C:\\Users\\UserName\\Desktop\\IMG\\" << i << ".png";
            fname = ssm.str();
            img[i] = cvLoadImage(fname.c_str(), CV_LOAD_IMAGE_GRAYSCALE);
            if( !img[i] ){
                std::cerr << "Failed : " << fname << std::endl;
                exit(EXIT_FAILURE);
            }
            std::cout << "Loaded : " << fname << std::endl;
            ssm.str("");
        }
    }

    // 比較元とする画像を指定
    do{
        std::cout << "Enter the image number of origin.\n";
        std::cin >> i;
    }while( ( i < 0 || num <= i ) && ( puts("The number exceeds the range.") || true ) );

    const int origin(i);

    // cvMatchShapes関数を利用して形状を比較する
    double* result = new double[num];
    for(i = 0; i < num; ++i){
        result[i] = cvMatchShapes( img[origin], img[i], CV_CONTOURS_MATCH_I1);
    }

    // 結果を表示
    for(i = 0; i < num; ++i){
        std::cout << "I(" << i << "," << origin << ") = " << result[i] << std::endl;
    }

    // ベストマッチを探す
    int minIdx = static_cast<int>(0 == origin);
    double minResult = result[minIdx];
    for( i = 0; i < num; ++i){
        if( i == origin ) continue;
        if( result[i] < minResult ){
            minResult = result[i];
            minIdx = i;
        }
    }
    std::cout << "Min : Image No." << minIdx << std::endl;

    // ウィンドウを開く
    cvNamedWindow("Original", CV_WINDOW_AUTOSIZE);
    cvNamedWindow("Best", CV_WINDOW_AUTOSIZE);

    // 画像表示
    cvShowImage("Original", img[origin]);
    cvShowImage("Best", img[minIdx]);

    cvWaitKey();

    // 破棄
    cvDestroyAllWindows();
    delete [] result;
    for(i = 0; i < num; ++i){
        cvReleaseImage( &img[i] );
    }
    return 0;
}