1

我有一个获取 ORB 关键点和图像描述符的代码，然后它使用“ BOWKMeansTrainer ”训练一个词袋。此代码的目的是使用 k-means 对照片进行聚类。我使用 ORB 没有任何问题。

这是我的问题：

我看到 openCV 提供了一个 VGG，你可以在这里找到它。我想使用 VGG 描述符而不是 ORB。根据文档，VGG 没有任何生成关键点的功能，所以我使用 ORB 关键点，然后使用 ORB 关键点创建 VGG 描述符。

Ptr<ORB> detector = ORB::create();
Ptr<cv::xfeatures2d::VGG> detectorVGG = cv::xfeatures2d::VGG::create(VGG::VGG_120, 1.4f, true, true, 0.75f, true);

detector->detect(grayFrame, kp);
detectorVGG->compute(grayFrame, kp, descriptors);

这一个也有效，我可以轻松地训练词袋并将它们聚类，就像我使用 ORB 的方式一样。但是，当我想测试词袋时，我收到了这个错误：

error: (-215) _queryDescriptors.type() == trainDescType in function knnMatchImpl   Aborttrap: 6

我试图更改查询描述符 Mat 和描述符 Mat 的类型，但我无法解决它。在最好的情况下，错误变为：

(-215) (type == CV_8U && dtype == CV_32S) || dtype == CV_32F in function batchDistance

我不确定问题是类型上的一个小错误，还是因为同时使用了 ORB 关键点和 VGG 描述符。

我的部分代码在这里：

vector<Mat> makeDescriptor(vector<Mat> frames)
    {
        int frameSize = frames.size();
        vector<Mat> characteristics;
        for (int i = 0 ; i < frameSize ; i++)
        {
            vector<KeyPoint> kp;
            Mat descriptors;
            Ptr<cv::xfeatures2d::VGG> detectorVGG = cv::xfeatures2d::VGG::create();
            Ptr<ORB> detectorORB = ORB::create();
            Mat frame = frames[i];
            if (frame.empty()) {
                cout << "Frame empty"<< "\n";
                continue;
            }
            Mat grayFrame;
            try
            {
                cvtColor(frame,grayFrame,CV_RGB2GRAY);
            }
            catch(Exception& e) 
            {
                continue;
            }

            detectorORB->detect(grayFrame,kp);
            detectorVGG->compute(grayFrame,kp,descriptors);
            characteristics.push_back(descriptors);
            }
            return characteristics;
        }

Mat makeDict(vector<Mat> characteristics, int _nCluster)
            {
                BOWKMeansTrainer bow(_nCluster);
                for (int j = 0 ; j < characteristics.size() ; j++)
                {
                    Mat descr = characteristics[j];
                    if (!descr.empty())
                    {
                        bow.add(descr);
                    }
                }
                Mat voc = bow.cluster();    
                return voc;
            }
BOWImgDescriptorExtractor makeBow(Mat dict) 
{

    Ptr<DescriptorMatcher> matcher = DescriptorMatcher::create("BruteForce-Hamming");
    Ptr<DescriptorExtractor> extractor = VGG::create();
    BOWImgDescriptorExtractor bowDE(extractor,matcher);
    Mat voc;
    dict.convertTo(voc,CV_32F);
    bowDE.setVocabulary(voc);
    return bowDE;
}

void testBow(BOWImgDescriptorExtractor bowDE, vector<Mat> frames)
{
    Mat features;
    vector< vector<KeyPoint> > keypoints = makeORBKeyPoints(frames);
    for (int i = 0 ; i < frames.size() ; i++) 
    {
        Mat bowDesc;

        if (!keypoints[i].empty())
        {
            cout << "inside if" << '\n';
            Mat frame;
            frames[i].convertTo(frame, CV_8U);
            bowDE.compute(frame, keypoints[i], bowDesc); //This Line
            bowDesc.convertTo(bowDesc,CV_32F);
        }
        features.push_back(bowDesc);
    }
}

运行时错误发生在这一行：

bowDE.compute(frame, keypoints[i], bowDesc); //This Line

如果有人帮助我找到解决方案，我将不胜感激。

谢谢，

4

0 回答 0