c++ - 我的应用程序在KNearest::find_nearest崩溃

我想实现OCR功能。

我已经收集了一些样本，我想使用K-Nearest来实现它。

因此，我使用以下代码加载数据并初始化KNearest

KNearest knn = new KNearest;
Mat mData, mClass;
for (int i = 0; i <= 9; ++i)
{
    Mat mImage = imread( FILENAME ); // the filename format is '%d.bmp', presenting a 15x15 image
    Mat mFloat;
    if (mImage.empty()) break; // if the file doesn't exist

    mImage.convertTo(mFloat, CV_32FC1);
    mData.push_back(mFloat.reshape(1, 1));
    mClass.push_back( '0' + i );
}

knn->train(mData, mClass);

然后，我调用代码以找到最佳结果

for (vector<Mat>::iterator it = charset.begin(); it != charset.end(); ++it)
{
    Mat mFloat;
    it->convertTo(mFloat, CV_32FC1); // 'it' presents a 15x15 gray image
    float result = knn->find_nearest(mFloat.reshape(1, 1), knn->get_max_k());
}

但是，我的应用程序在find_nearest崩溃。

有人可以帮助我吗？

最佳答案

我似乎找到了问题...

我的样本图像是cvtColor转换后的灰度图像，但我的输入图像不是。

我添加后

cvtColor(mImage, mImage, COLOR_BGR2GRAY);

之间

if (mImage.empty()) break;
mImage.convertTo(mFloat, CV_32FC1);

find_nearest()返回一个值，我的应用程序正常。