web-dev-qa-db-ja.com

OpenCV：アサーションに失敗しました（（img.depth（）== CV_8U || img.depth（）== CV_32F）&& img.type（）== templ.type（））

上からこのエラーが発生し、回避する方法がわかりません。私の目的は、スクリーンショットを取得し、それに一致するテンプレートを実行して、現時点でアイコンが画面に表示されているかどうかを確認することです。今までは、アイコンの位置だけでした。私のコード：

#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/core/core.hpp"
#include <Windows.h>
#include <iostream>
#include <stdio.h>
#include <string>


using namespace std;
using namespace cv;

Mat hwnd2mat();

/// Global Variables
Mat img; Mat templ; Mat result;

int main()
{
  /// Load image and template
  templ = imread( "Template.bmp",1);
  templ.convertTo(templ, CV_8U);

  //img = imread( "Image.jpg", 1 );
  img = hwnd2mat();


  /// Create the result matrix
  int result_cols =  img.cols - templ.cols + 1;
  int result_rows = img.rows - templ.rows + 1;

  result.create( result_cols, result_rows, CV_8U);

  /// Do the Matching and Normalize
  matchTemplate( img, templ, result, CV_TM_SQDIFF );
  normalize( result, result, 0, 1, NORM_MINMAX, -1, Mat() );

  /// Localizing the best match with minMaxLoc
  double minVal; double maxVal; Point minLoc; Point maxLoc;
  Point matchLoc;

  minMaxLoc( result, &minVal, &maxVal, &minLoc, &maxLoc, Mat() );

  /// show best position
  matchLoc = minLoc;
  cout<<matchLoc<<" is best position"<<endl;

  waitKey(0);
  return 0;
}

Mat hwnd2mat(){

    HWND hwnd = GetDesktopWindow();
    HDC hwindowDC,hwindowCompatibleDC;

    int height,width,srcheight,srcwidth;
    HBITMAP hbwindow;
    Mat src;
    BITMAPINFOHEADER  bi;

    hwindowDC=GetDC(hwnd);
    hwindowCompatibleDC=CreateCompatibleDC(hwindowDC);
    SetStretchBltMode(hwindowCompatibleDC,COLORONCOLOR);  

    RECT windowsize;    // get the height and width of the screen
    GetClientRect(hwnd, &windowsize);

    srcheight = windowsize.bottom;
    srcwidth = windowsize.right;
    height = windowsize.bottom/1;  //change this to whatever size you want to resize to
    width = windowsize.right/1;

    src.create(height,width,CV_8U);

    // create a bitmap
    hbwindow = CreateCompatibleBitmap( hwindowDC, width, height);
    bi.biSize = sizeof(BITMAPINFOHEADER);
    bi.biWidth = width;    
    bi.biHeight = -height;  //this is the line that makes it draw upside down or not
    bi.biPlanes = 1;    
    bi.biBitCount = 32;    
    bi.biCompression = BI_RGB;    
    bi.biSizeImage = 0;  
    bi.biXPelsPerMeter = 0;    
    bi.biYPelsPerMeter = 0;    
    bi.biClrUsed = 0;    
    bi.biClrImportant = 0;

    // use the previously created device context with the bitmap
    SelectObject(hwindowCompatibleDC, hbwindow);
    // copy from the window device context to the bitmap device context
    StretchBlt( hwindowCompatibleDC, 0,0, width, height, hwindowDC, 0, 0,srcwidth,srcheight, SRCCOPY); //change SRCCOPY to NOTSRCCOPY for wacky colors !
    GetDIBits(hwindowCompatibleDC,hbwindow,0,height,src.data,(BITMAPINFO *)&bi,DIB_RGB_COLORS);  //copy from hwindowCompatibleDC to hbwindow

    DeleteObject(hbwindow);
    DeleteDC(hwindowCompatibleDC);
    ReleaseDC(hwnd, hwindowDC);

    return src;
}

スクリーンショット付きの関数は私自身の作品ではありません、私はそれを ここ から取得しました

何をすべきかアイデアはありますか？

助けてくれてありがとう、よろしく！

c++opencvdepth
12
Bl4cke4gle

問題は、関数_hwnd2mat_がタイプ_CV_8UC1_のグレースケール画像を返すのに対し、templはタイプ_CV_8UC3_のカラー画像を返すことです。したがって、失敗した条件img.type() == templ.type()が原因で、関数matchTemplateでアサーションが失敗しています。エラーを回避するために、画像をグレースケールとしてロードすることができます。

_templ = imread( "Template.bmp",CV_LOAD_IMAGE_GRAYSCALE);
_

更新：

関数_hwnd2mat_が現在の形式で機能しておらず、無効な画像を返すことに注意してください。元のコードは、正しいアプローチであるタイプ_CV_8UC4_の出力イメージを作成します。

_src.create(height,width,CV_8UC4);
_

_hwnd2mat_から戻る前にsrcをグレースケールに変換するか、templを4チャンネル画像に変換することができます。とにかく、ポイントは、matchTemplateが機能するには、両方の画像が同じタイプである必要があるということです。

7
sgarizvi