Opencv之形状距离匹配--createShapeContextDistanceExtractor

代码展示:

c++版:

#include <iostream>
#include <string>
#include <opencv2/shape.hpp>
#include <opencv2/opencv.hpp>
using namespace std;
using namespace cv;


static vector<Point> simpleContour( const Mat& currentQuery, int n=300 )
{
    vector<vector<Point> > _contoursQuery;
    vector <Point> contoursQuery;

    findContours(currentQuery, _contoursQuery, RETR_LIST, CHAIN_APPROX_NONE);
    for (size_t border=0; border<_contoursQuery.size(); border++)
    {
        for (size_t p=0; p<_contoursQuery[border].size(); p++)
        {
            contoursQuery.push_back( _contoursQuery[border][p] );
        }
    }

    // 如果实际点数小于n
    int dummy=0;
    for (int add=(int)contoursQuery.size()-1; add<n; add++)
    {

        contoursQuery.push_back(contoursQuery[dummy++]); //adding dummy values
    }
    cout << contoursQuery.size() <<endl;
    // 提取前n个点
    cv::randShuffle(contoursQuery);
    vector<Point> cont;
    for (int i=0; i<n; i++)
    {
        cont.push_back(contoursQuery[i]);
    }

    return cont;
}

int main(int argc, char** argv)
{
    string path = "/home/pi/ProgramFiles/opencv4/opencv_contrib/modules/shape/samples/data/shape_sample/";
    int indexQuery = 1;

    cv::Ptr <cv::ShapeContextDistanceExtractor> mysc = cv::createShapeContextDistanceExtractor();
    Size sz2Sh(300,300);
    stringstream queryName;
    queryName<<path<<indexQuery<<".png";
    Mat query=imread(queryName.str(), IMREAD_GRAYSCALE);
    Mat queryToShow;
    resize(query, queryToShow, sz2Sh, 0, 0, INTER_LINEAR_EXACT);
    imshow("QUERY", queryToShow);
    moveWindow("TEST", 0,0);
    vector<Point> contQuery = simpleContour(query);
    int bestMatch = 0;
    float bestDis=FLT_MAX;
    for ( int ii=1; ii<=20; ii++ )
    {
        if (ii==indexQuery) continue;
        waitKey(30);
        stringstream iiname;
        iiname<<path<<ii<<".png";
        cout<<"name: "<<iiname.str()<<endl;
        Mat iiIm=imread(iiname.str(), 0);
        Mat iiToShow;
        resize(iiIm, iiToShow, sz2Sh, 0, 0, INTER_LINEAR_EXACT);
        imshow("TEST", iiToShow);
        moveWindow("TEST", sz2Sh.width+50,0);
        vector<Point> contii = simpleContour(iiIm);
        float dis = mysc->computeDistance( contQuery, contii );
        if ( dis<bestDis )
        {
            bestMatch = ii;
            bestDis = dis;
        }
        std::cout<<" distance between "<<queryName.str()<<" and "<<iiname.str()<<" is: "<<dis<<std::endl;
    }
    destroyWindow("TEST");
    stringstream bestname;
    bestname<<path<<bestMatch<<".png";
    Mat iiIm=imread(bestname.str(), 0);
    Mat bestToShow;
    resize(iiIm, bestToShow, sz2Sh, 0, 0, INTER_LINEAR_EXACT);
    imshow("BEST MATCH", bestToShow);
    moveWindow("BEST MATCH", sz2Sh.width+50,0);
    waitKey();

    return 0;
}

python 版:

# -*- coding:utf-8 -*-
import os
import cv2
import random
import numpy as np


def simple_contour(current_query_image, n=300):
    contours_query_list = []
    contours = cv2.findContours(current_query_image.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)[-2]
    for one_contour in contours:
        for one_point in one_contour:
            contours_query_list.append(one_point)
    # 如果实际点数小于n
    dummy = 0
    for i in range(len(contours) - 1, n):
        contours_query_list.append(contours_query_list[dummy])
        dummy += 1
    random.shuffle(contours_query_list)
    cont_list = []
    for i in range(n):
        cont_list.append(contours_query_list[i])

    return np.array(cont_list)


image_path = "/home/pi/ProgramFiles/opencv4/opencv_contrib/modules/shape/samples/data/shape_sample/"
query_index = 1

dsize = (300, 300)
query_image = cv2.imread(os.path.join(image_path, "{}.png".format(query_index)), cv2.IMREAD_GRAYSCALE)
query_image = cv2.resize(query_image, dsize)
cv2.imshow('query_image', query_image)

contours_query_list = simple_contour(query_image)
mysc = cv2.createShapeContextDistanceExtractor()
best_match = 0
best_distance = np.inf

for image_index in range(1, 21):
    if image_index == query_index:
        continue
    matched_image = cv2.imread(os.path.join(image_path, "{}.png".format(image_index)), cv2.IMREAD_GRAYSCALE)
    matched_image = cv2.resize(matched_image, dsize)
    cv2.imshow('test', matched_image)
    cv2.waitKey(33)
    contours_matched_list = simple_contour(matched_image)
    distance = mysc.computeDistance(contours_query_list, contours_matched_list)
    if distance < best_distance:
        best_match = image_index
        best_distance = distance

cv2.destroyWindow('test')
best_match_image = cv2.imread(os.path.join(image_path, "{}.png".format(best_match)), 0)
cv2.imshow('best_image', best_match_image)

cv2.waitKey()

效果展示: 

测试数据: 下载地址