Skip to content

paulyehtw/SFND_2D_Feature_Tracking

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SFND 2D Feature Tracking

The idea of the camera course is to build a collision detection system - that's the overall goal for the Final Project. As a preparation for this, you will now build the feature tracking part and test various detector / descriptor combinations to see which ones perform best. This mid-term project consists of four parts:

  • First, you will focus on loading images, setting up data structures and putting everything into a ring buffer to optimize memory load.
  • Then, you will integrate several keypoint detectors such as HARRIS, FAST, BRISK and SIFT and compare them with regard to number of keypoints and speed.
  • In the next part, you will then focus on descriptor extraction and matching using brute force and also the FLANN approach we discussed in the previous lesson.
  • In the last part, once the code framework is complete, you will test the various algorithms in different combinations and compare them with regard to some performance measures.

See the classroom instruction and code comments for more details on each of these parts. Once you are finished with this project, the keypoint matching part will be set up and you can proceed to the next lesson, where the focus is on integrating Lidar points and on object detection using deep-learning.

Dependencies for Running Locally

Basic Build Instructions

  1. Clone this repo.
  2. Make a build directory in the top level directory: mkdir build && cd build
  3. Compile: cmake .. && make
  4. Run it: ./2D_feature_tracking.

Writeup for submission

Data Buffer

MP.1 Data Buffer Optimization

in MidTermProject_Camera_Student.cpp : implement ring buffer logic

if (dataBuffer.size() < dataBufferSize)
{
    dataBuffer.push_back(frame);
}
else
{
    dataBuffer.erase(dataBuffer.begin());
    dataBuffer.push_back(frame);
}

Keypoints

MP.2 Keypoint Detection

in MidTermProject_Camera_Student.cpp : make detectorType selectable by setting a string accordingly

if (detectorType == "SHITOMASI")
{
    detKeypointsShiTomasi(keypoints, imgGray, false);
}
else if (detectorType == "HARRIS")
{
    detKeypointsHarris(keypoints, imgGray, false);
}
else if (detectorType == "FAST" ||
         detectorType == "BRISK" ||
         detectorType == "ORB" ||
         detectorType == "AKAZE" ||
         detectorType == "SIFT")
{
    detKeypointsModern(keypoints, imgGray, detectorType, false);
}
else
{
    // Default keypoint detector
    cout << "\033[1;33mNo keypoint detector is seletecd, using the Shi-Tomasi detector as default\033[0m\n";
    detKeypointsShiTomasi(keypoints, imgGray, false);
}

in matching2D_Student.cpp : implement detKeypointsHarris

void detKeypointsHarris(vector<cv::KeyPoint> &keypoints, cv::Mat &img, bool bVis)
{
    // Set detector parameters
    int blockSize = 2;         // Neighborhood size
    int apertureSize = 3;      // Aperture parameter for the Sobel operator.
    double k = 0.04;           // Harris detector free parameter
    int cornerThreshold = 100; // Threshold for being considered as a corner

    double t = (double)cv::getTickCount();
    vector<cv::Point2f> corners;
    // Apply corner detection
    cv::Mat dst = cv::Mat::zeros(img.size(), CV_32FC1);
    cv::cornerHarris(img, dst, blockSize, apertureSize, k);

    // Normalize the result
    cv::Mat dst_norm;
    cv::normalize(dst, dst_norm, 0, 255, cv::NORM_MINMAX, CV_32FC1, cv::Mat());

    // Add detected corners to the result vector
    for (int i = 0; i < dst_norm.rows; i++)
    {
        for (int j = 0; j < dst_norm.cols; j++)
        {
            if ((int)dst_norm.at<float>(i, j) > cornerThreshold)
            {
                cv::KeyPoint newKeyPoint;
                newKeyPoint.pt = cv::Point2f(j, i);
                newKeyPoint.size = blockSize;
                keypoints.push_back(newKeyPoint);
            }
        }
    }
    t = ((double)cv::getTickCount() - t) / cv::getTickFrequency();
    cout << "Harris detection with n=" << keypoints.size() << " keypoints in " << 1000 * t / 1.0 << " ms" << endl;

    // visualize results
    if (bVis)
    {
        cv::Mat visImage = img.clone();
        cv::drawKeypoints(img, keypoints, visImage, cv::Scalar::all(-1), cv::DrawMatchesFlags::DRAW_RICH_KEYPOINTS);
        string windowName = "Harris Corner Detector Results";
        cv::namedWindow(windowName, 6);
        imshow(windowName, visImage);
        cv::waitKey(0);
    }
}

in matching2D_Student.cpp : implement detKeypointsModern for FAST, BRISK, ORB, AKAZE and SIFT

void detKeypointsModern(std::vector<cv::KeyPoint> &keypoints, cv::Mat &img, std::string detectorType, bool bVis)
{
    double t = (double)cv::getTickCount();
    // Apply corner detection
    if (detectorType == "FAST")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::FastFeatureDetector::create();
        detector->detect(img, keypoints);
    }
    if (detectorType == "BRISK")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::BRISK::create();
        detector->detect(img, keypoints);
    }
    else if (detectorType == "ORB")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::ORB::create();
        detector->detect(img, keypoints);
    }
    else if (detectorType == "AKAZE")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::AKAZE::create();
        detector->detect(img, keypoints);
    }
    else if (detectorType == "SIFT")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::SIFT::create();
        detector->detect(img, keypoints);
    }

    t = ((double)cv::getTickCount() - t) / cv::getTickFrequency();
    cout << detectorType + " detection with n=" << keypoints.size() << " keypoints in " << 1000 * t / 1.0 << " ms" << endl;

    // visualize results
    if (bVis)
    {
        cv::Mat visImage = img.clone();
        cv::drawKeypoints(img, keypoints, visImage, cv::Scalar::all(-1), cv::DrawMatchesFlags::DRAW_RICH_KEYPOINTS);
        string windowName = detectorType + " Detector Results";
        cv::namedWindow(windowName, 6);
        imshow(windowName, visImage);
        cv::waitKey(0);
    }

MP.3 Keypoint Removal

in MidTermProject_Camera_Student.cpp : implement cropping in main

if (bFocusOnVehicle)
{
    vector<cv::KeyPoint> croppedKeypoints; // create empty list for cropped keypoints
    for (cv::KeyPoint keypoint : keypoints)
    {
        if ((keypoint.pt.x >= vehicleRect.x && keypoint.pt.x <= vehicleRect.x + vehicleRect.width) &&
            (keypoint.pt.y >= vehicleRect.y && keypoint.pt.y <= vehicleRect.y + vehicleRect.height))
        {
            croppedKeypoints.push_back(keypoint);
        }
    }
    keypoints = croppedKeypoints;
}

Descriptors

MP.4 Keypoint Descriptors

in matching2D_Student.cpp : Implement descriptors BRIEF, ORB, FREAK, AKAZE and SIFT

void detKeypointsModern(std::vector<cv::KeyPoint> &keypoints, cv::Mat &img, std::string detectorType, bool bVis)
{
    double t = (double)cv::getTickCount();
    // Apply corner detection
    if (detectorType == "FAST")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::FastFeatureDetector::create();
        detector->detect(img, keypoints);
    }
    else if (detectorType == "BRISK")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::BRISK::create();
        detector->detect(img, keypoints);
    }
    else if (detectorType == "ORB")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::ORB::create();
        detector->detect(img, keypoints);
    }
    else if (detectorType == "AKAZE")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::AKAZE::create();
        detector->detect(img, keypoints);
    }
    else if (detectorType == "SIFT")
    {
        cv::Ptr<cv::FeatureDetector> detector = cv::SIFT::create();
        detector->detect(img, keypoints);
    }

    t = ((double)cv::getTickCount() - t) / cv::getTickFrequency();
    cout << detectorType + " detection with n=" << keypoints.size() << " keypoints in " << 1000 * t / 1.0 << " ms" << endl;

    // visualize results
    if (bVis)
    {
        cv::Mat visImage = img.clone();
        cv::drawKeypoints(img, keypoints, visImage, cv::Scalar::all(-1), cv::DrawMatchesFlags::DRAW_RICH_KEYPOINTS);
        string windowName = detectorType + " Detector Results";
        cv::namedWindow(windowName, 6);
        imshow(windowName, visImage);
        cv::waitKey(0);
    }
}

MP.5 Descriptor Matching

in matching2D_Student.cpp : Implement FLANN and KNN by adding followings in matchDescriptors

if (matcherType == "MAT_BF")
{
    int normType = cv::NORM_HAMMING;
    matcher = cv::BFMatcher::create(normType, crossCheck);
}
else if (matcherType == "MAT_FLANN")
{
    // In order to use FlannBasedMatcher, descriptors need to be CV_32F
    if (descSource.type() != CV_32F)
    {
        descSource.convertTo(descSource, CV_32F);
    }
    if (descRef.type() != CV_32F)
    {
        descRef.convertTo(descRef, CV_32F);
    }
    matcher = cv::FlannBasedMatcher::create();
}

// perform matching task
if (selectorType == "SEL_NN")
{
    // nearest neighbor (best match)
    matcher->match(descSource, descRef, matches); // Finds the best match for each descriptor in desc1
}
else if (selectorType == "SEL_KNN")
{
    // k nearest neighbors (k=2)
    int k = 2;
    float distanceRatioThreshold = 0.8F;
    std::vector<std::vector<cv::DMatch>> knnMatches;
    matcher->knnMatch(descSource, descRef, knnMatches, k);
    for (auto match : knnMatches)
    {
        if (match[0].distance < distanceRatioThreshold * match[1].distance)
        {
            matches.push_back(match[0]);
        }
    }
}

MP.6 Descriptor Distance Ratio

in matching2D_Student.cpp : Implement distance ratio in for KNN(k=2)

int k = 2;
float distanceRatioThreshold = 0.8F;
std::vector<std::vector<cv::DMatch>> knnMatches;
matcher->knnMatch(descSource, descRef, knnMatches, k);
for (auto match : knnMatches)
{
    if (match[0].distance < distanceRatioThreshold * match[1].distance)
    {
        matches.push_back(match[0]);
    }
}

Performance

Please checkout Performance_Evaluation to run the evaluation

MP.7 Performance Evaluation 1

Number of detected keypoints for each detector on the preceding vehicle for total of 10 images

Detector Number of detected Keypoints
SHITOMASI 1189
HARRIS 1084
FAST 4136
BRISK 2714
ORB 1150
AKAZE 1655
SIFT 1371

MP.8 Performance Evaluation 2

The number of matched keypoints for all 10 images using all possible combinations of detectors and descriptors

Detector\Descriptor BRISK BRIEF ORB FREAK AKAZE SIFT
SHITOMASI 771 953 914 768 N/A 936
HARRIS 393 423 412 443 N/A 409
FAST 2248 2924 2869 2284 N/A 2805
BRISK 1563 1805 1496 1518 N/A 1620
ORB 753 576 757 438 N/A 763
AKAZE 1206 1322 1186 1183 1249 1263
SIFT 602 751 Out of Memory 626 N/A 820

MP.9 Performance Evaluation 3

Average Processing Time (ms) on all images for each detector/descriptor combination

Detector\Descriptor BRISK BRIEF ORB FREAK AKAZE SIFT
SHITOMASI 601.261 25.3585 34.3536 104.04 N/A 63.8929
HARRIS 564.416 28.8252 35.3673 103.321 N/A 59.392
FAST 554.408 11.0708 17.3062 88.828 N/A 75.9405
BRISK 1174.21 616.871 645.412 691.046 N/A 702.332
ORB 547.948 20.8855 50.7708 100.113 N/A 112.002
AKAZE 745.616 215.533 232.859 286.765 344.428 257.345
SIFT 762.529 227.408 Out of Memory 326.65 N/A 343.704

TOP 3 suggestion

Given that we are not looking into the correctness of matched points, I rank the performance mainly based on processing time. The combination that runs faster while getting fairly good amount of matched points is ranked higher. Top 3 combinations are ranked as following:

Rank Combination Processing Time(ms) Matched Points
1 FAST/BRIEF 11.0708 2924
2 FAST/ORB 17.3062 2869
3 ORB/ORB 50.7708 757

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 92.9%
  • Shell 4.5%
  • CMake 2.6%