c++ - opencv stereoRectifyUncalibrated 的问题

标签 c++ opencv stereo-3d disparity-mapping

我一直在尝试使用 OpenCV stereoRectifyUncalibrated 纠正和构建一对图像的视差映射,但我没有得到很好的结果。我的代码是:

template<class T>
T convertNumber(string& number)
{
    istringstream ss(number);
    T t;
    ss >> t;
    return t;
}

void readPoints(vector<Point2f>& points, string filename)
{
    fstream filest(filename.c_str(), ios::in);
    string line;

    assert(filest != NULL);

    getline(filest, line);
    do{
        int posEsp = line.find_first_of(' ');
        string posX = line.substr(0, posEsp);
        string posY = line.substr(posEsp+1, line.size() - posEsp);

        float X = convertNumber<float>(posX);
        float Y = convertNumber<float>(posY);

        Point2f pnt = Point2f(X, Y);
        points.push_back(pnt);
        getline(filest, line);
    }while(!filest.eof());

    filest.close();
}

void drawKeypointSequence(Mat lFrame, Mat rFrame, vector<KeyPoint>& lKeyp, vector<KeyPoint>& rKeyp)
{
    namedWindow("prevFrame", WINDOW_AUTOSIZE);
    namedWindow("currFrame", WINDOW_AUTOSIZE);
    moveWindow("prevFrame", 0, 300);
    moveWindow("currFrame", 650, 300);
    Mat rFrameAux;
    rFrame.copyTo(rFrameAux);
    Mat lFrameAux;
    lFrame.copyTo(lFrameAux);

    int size = rKeyp.size();
    for(int i=0; i<size; i++)
    {
        vector<KeyPoint> drawRightKeyp;
        vector<KeyPoint> drawleftKeyp;

        drawRightKeyp.push_back(rKeyp[i]);
        drawleftKeyp.push_back(lKeyp[i]);

        cout << rKeyp[i].pt << " <<<>>> " << lKeyp[i].pt << endl;

        drawKeypoints(rFrameAux, drawRightKeyp, rFrameAux, Scalar::all(255), DrawMatchesFlags::DRAW_OVER_OUTIMG);
        drawKeypoints(lFrameAux, drawleftKeyp, lFrameAux, Scalar::all(255), DrawMatchesFlags::DRAW_OVER_OUTIMG);

        imshow("currFrame", rFrameAux);
        imshow("prevFrame", lFrameAux);
        waitKey(0);
    }
    imwrite("RightKeypFrame.jpg", rFrameAux);
    imwrite("LeftKeypFrame.jpg", lFrameAux);
}
int main(int argc, char* argv[])
{
    StereoBM stereo(StereoBM::BASIC_PRESET, 16*5, 21);
    double ndisp = 16*4;
    assert(argc == 5);
    string rightImgFilename(argv[1]);       // Right image (current frame)
    string leftImgFilename(argv[2]);        // Left image (previous frame)
    string rightPointsFilename(argv[3]);    // Right image points file
    string leftPointsFilename(argv[4]);     // Left image points file

    Mat rightFrame = imread(rightImgFilename.c_str(), 0);
    Mat leftFrame = imread(leftImgFilename.c_str(), 0);

    vector<Point2f> rightPoints;
    vector<Point2f> leftPoints;

    vector<KeyPoint> rightKeyp;
    vector<KeyPoint> leftKeyp;

    readPoints(rightPoints, rightPointsFilename);
    readPoints(leftPoints, leftPointsFilename);
    assert(rightPoints.size() == leftPoints.size());

    KeyPoint::convert(rightPoints, rightKeyp);
    KeyPoint::convert(leftPoints, leftKeyp);

    // Desenha os keypoints sequencialmente, de forma a testar a consistência do matching
    drawKeypointSequence(leftFrame, rightFrame, leftKeyp, rightKeyp);

    Mat fundMatrix = findFundamentalMat(leftPoints, rightPoints, CV_FM_8POINT);
    Mat homRight;
    Mat homLeft;
    Mat disp16 = Mat(rightFrame.rows, leftFrame.cols, CV_16S);
    Mat disp8 = Mat(rightFrame.rows, leftFrame.cols, CV_8UC1);
    stereoRectifyUncalibrated(leftPoints, rightPoints, fundMatrix, rightFrame.size(), homLeft, homRight);

    warpPerspective(rightFrame, rightFrame, homRight, rightFrame.size());
    warpPerspective(leftFrame, leftFrame, homLeft, leftFrame.size());


    namedWindow("currFrame", WINDOW_AUTOSIZE);
    namedWindow("prevFrame", WINDOW_AUTOSIZE);
    moveWindow("currFrame", 650, 300);
    moveWindow("prevFrame", 0, 300);
    imshow("currFrame", rightFrame);
    imshow("prevFrame", leftFrame);

    imwrite("RectfRight.jpg", rightFrame);
    imwrite("RectfLeft.jpg", leftFrame);

    waitKey(0);

    stereo(rightFrame, leftFrame, disp16, CV_16S);

    disp16.convertTo(disp8, CV_8UC1, 255/ndisp);
    FileStorage file("disp_map.xml", FileStorage::WRITE);
    file << "disparity" << disp8;
    file.release();
    imshow("disparity", disp8);
    imwrite("disparity.jpg", disp8);
    moveWindow("disparity", 0, 0);

    waitKey(0);
}

drawKeyPoint 序列是我目视检查两个图像的点的一致性的方式。通过按顺序绘制它们的每个关键点,我可以确定图像 A 上的关键点 i 是图像 B 上的关键点 i。

我也试过使用 ndisp 参数,但没有太大帮助。

我尝试了以下一对图像:

LeftImage

RightImage

得到以下修正对:

RectifiedLeft

RectifiedRight

最后是下面的视差图

DisparityMap

如您所见,这非常糟糕。我还使用以下 stereoRectifyUncalibrated 示例尝试了同一对图像:http://programmingexamples.net/wiki/OpenCV/WishList/StereoRectifyUncalibrated和来自 opencv 教程代码示例的 SBM_Sample.cpp 构建视差图,并得到非常相似的结果。

我正在使用 opencv 2.4

提前致谢!

最佳答案

除了可能存在的校准问题外,您的图像显然缺少立体 block 匹配无法正常工作的某些纹理。 该算法会在平坦(非纹理化)部分看到许多歧义和过大的差异。

但请注意,关键点似乎匹配得很好,所以即使校正输出看起来很奇怪,它也可能是正确的。

您可以根据 Middlebury stereo page 中的标准图像测试您的代码用于完整性检查。

关于c++ - opencv stereoRectifyUncalibrated 的问题,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/11190732/

相关文章:

c++ - 如何仅在调用函数B之后才允许调用函数A?

c++ - 为什么我们两次使用引用运算符?

c++ - 从 BruteForceMatcher_GPU 调用 match 方法时出现编译错误

opencv - 当相机仅向前/向后移动时如何找到视差

python-2.7 - Python 2.7:cv2.stereoCalibrate()中的错误。找不到必需的参数 'distCoeffs1'(pos 5)

c++ - 需要对 "High definition time-stamps"(win32、direct3d、性能计数器)进行一些说明

c++ - void QTableWidget::setItemPrototype ( const QTableWidgetItem * item ) 如何克隆对象?

opencv - 如何在 skimage 中执行 cv2.min 操作?

c++ - 如何从图像中分割出真实的人体形状?

python - 使用 OpenCV 和 Python 进行立体校正的问题