Renesas / opencv-lib

openCV library for Renesas RZ/A

Dependents:   RZ_A2M_Mbed_samples

include/opencv2/flann/ground_truth.h@0:0e0631af0305, 2021-01-29 (annotated)

Committer:
RyoheiHagimoto
Date:
Fri Jan 29 04:53:38 2021 +0000
Revision:
0:0e0631af0305
copied from https://github.com/d-kato/opencv-lib.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
RyoheiHagimoto 0:0e0631af0305 1 /***********************************************************************
RyoheiHagimoto 0:0e0631af0305 2 * Software License Agreement (BSD License)
RyoheiHagimoto 0:0e0631af0305 3 *
RyoheiHagimoto 0:0e0631af0305 4 * Copyright 2008-2009 Marius Muja (mariusm@cs.ubc.ca). All rights reserved.
RyoheiHagimoto 0:0e0631af0305 5 * Copyright 2008-2009 David G. Lowe (lowe@cs.ubc.ca). All rights reserved.
RyoheiHagimoto 0:0e0631af0305 6 *
RyoheiHagimoto 0:0e0631af0305 7 * THE BSD LICENSE
RyoheiHagimoto 0:0e0631af0305 8 *
RyoheiHagimoto 0:0e0631af0305 9 * Redistribution and use in source and binary forms, with or without
RyoheiHagimoto 0:0e0631af0305 10 * modification, are permitted provided that the following conditions
RyoheiHagimoto 0:0e0631af0305 11 * are met:
RyoheiHagimoto 0:0e0631af0305 12 *
RyoheiHagimoto 0:0e0631af0305 13 * 1. Redistributions of source code must retain the above copyright
RyoheiHagimoto 0:0e0631af0305 14 * notice, this list of conditions and the following disclaimer.
RyoheiHagimoto 0:0e0631af0305 15 * 2. Redistributions in binary form must reproduce the above copyright
RyoheiHagimoto 0:0e0631af0305 16 * notice, this list of conditions and the following disclaimer in the
RyoheiHagimoto 0:0e0631af0305 17 * documentation and/or other materials provided with the distribution.
RyoheiHagimoto 0:0e0631af0305 18 *
RyoheiHagimoto 0:0e0631af0305 19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
RyoheiHagimoto 0:0e0631af0305 20 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
RyoheiHagimoto 0:0e0631af0305 21 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
RyoheiHagimoto 0:0e0631af0305 22 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
RyoheiHagimoto 0:0e0631af0305 23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
RyoheiHagimoto 0:0e0631af0305 24 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
RyoheiHagimoto 0:0e0631af0305 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
RyoheiHagimoto 0:0e0631af0305 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
RyoheiHagimoto 0:0e0631af0305 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
RyoheiHagimoto 0:0e0631af0305 28 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
RyoheiHagimoto 0:0e0631af0305 29 *************************************************************************/
RyoheiHagimoto 0:0e0631af0305 30
RyoheiHagimoto 0:0e0631af0305 31 #ifndef OPENCV_FLANN_GROUND_TRUTH_H_
RyoheiHagimoto 0:0e0631af0305 32 #define OPENCV_FLANN_GROUND_TRUTH_H_
RyoheiHagimoto 0:0e0631af0305 33
RyoheiHagimoto 0:0e0631af0305 34 #include "dist.h"
RyoheiHagimoto 0:0e0631af0305 35 #include "matrix.h"
RyoheiHagimoto 0:0e0631af0305 36
RyoheiHagimoto 0:0e0631af0305 37
RyoheiHagimoto 0:0e0631af0305 38 namespace cvflann
RyoheiHagimoto 0:0e0631af0305 39 {
RyoheiHagimoto 0:0e0631af0305 40
RyoheiHagimoto 0:0e0631af0305 41 template <typename Distance>
RyoheiHagimoto 0:0e0631af0305 42 void find_nearest(const Matrix<typename Distance::ElementType>& dataset, typename Distance::ElementType* query, int* matches, int nn,
RyoheiHagimoto 0:0e0631af0305 43 int skip = 0, Distance distance = Distance())
RyoheiHagimoto 0:0e0631af0305 44 {
RyoheiHagimoto 0:0e0631af0305 45 typedef typename Distance::ResultType DistanceType;
RyoheiHagimoto 0:0e0631af0305 46 int n = nn + skip;
RyoheiHagimoto 0:0e0631af0305 47
RyoheiHagimoto 0:0e0631af0305 48 std::vector<int> match(n);
RyoheiHagimoto 0:0e0631af0305 49 std::vector<DistanceType> dists(n);
RyoheiHagimoto 0:0e0631af0305 50
RyoheiHagimoto 0:0e0631af0305 51 dists[0] = distance(dataset[0], query, dataset.cols);
RyoheiHagimoto 0:0e0631af0305 52 match[0] = 0;
RyoheiHagimoto 0:0e0631af0305 53 int dcnt = 1;
RyoheiHagimoto 0:0e0631af0305 54
RyoheiHagimoto 0:0e0631af0305 55 for (size_t i=1; i<dataset.rows; ++i) {
RyoheiHagimoto 0:0e0631af0305 56 DistanceType tmp = distance(dataset[i], query, dataset.cols);
RyoheiHagimoto 0:0e0631af0305 57
RyoheiHagimoto 0:0e0631af0305 58 if (dcnt<n) {
RyoheiHagimoto 0:0e0631af0305 59 match[dcnt] = (int)i;
RyoheiHagimoto 0:0e0631af0305 60 dists[dcnt++] = tmp;
RyoheiHagimoto 0:0e0631af0305 61 }
RyoheiHagimoto 0:0e0631af0305 62 else if (tmp < dists[dcnt-1]) {
RyoheiHagimoto 0:0e0631af0305 63 dists[dcnt-1] = tmp;
RyoheiHagimoto 0:0e0631af0305 64 match[dcnt-1] = (int)i;
RyoheiHagimoto 0:0e0631af0305 65 }
RyoheiHagimoto 0:0e0631af0305 66
RyoheiHagimoto 0:0e0631af0305 67 int j = dcnt-1;
RyoheiHagimoto 0:0e0631af0305 68 // bubble up
RyoheiHagimoto 0:0e0631af0305 69 while (j>=1 && dists[j]<dists[j-1]) {
RyoheiHagimoto 0:0e0631af0305 70 std::swap(dists[j],dists[j-1]);
RyoheiHagimoto 0:0e0631af0305 71 std::swap(match[j],match[j-1]);
RyoheiHagimoto 0:0e0631af0305 72 j--;
RyoheiHagimoto 0:0e0631af0305 73 }
RyoheiHagimoto 0:0e0631af0305 74 }
RyoheiHagimoto 0:0e0631af0305 75
RyoheiHagimoto 0:0e0631af0305 76 for (int i=0; i<nn; ++i) {
RyoheiHagimoto 0:0e0631af0305 77 matches[i] = match[i+skip];
RyoheiHagimoto 0:0e0631af0305 78 }
RyoheiHagimoto 0:0e0631af0305 79 }
RyoheiHagimoto 0:0e0631af0305 80
RyoheiHagimoto 0:0e0631af0305 81
RyoheiHagimoto 0:0e0631af0305 82 template <typename Distance>
RyoheiHagimoto 0:0e0631af0305 83 void compute_ground_truth(const Matrix<typename Distance::ElementType>& dataset, const Matrix<typename Distance::ElementType>& testset, Matrix<int>& matches,
RyoheiHagimoto 0:0e0631af0305 84 int skip=0, Distance d = Distance())
RyoheiHagimoto 0:0e0631af0305 85 {
RyoheiHagimoto 0:0e0631af0305 86 for (size_t i=0; i<testset.rows; ++i) {
RyoheiHagimoto 0:0e0631af0305 87 find_nearest<Distance>(dataset, testset[i], matches[i], (int)matches.cols, skip, d);
RyoheiHagimoto 0:0e0631af0305 88 }
RyoheiHagimoto 0:0e0631af0305 89 }
RyoheiHagimoto 0:0e0631af0305 90
RyoheiHagimoto 0:0e0631af0305 91
RyoheiHagimoto 0:0e0631af0305 92 }
RyoheiHagimoto 0:0e0631af0305 93
RyoheiHagimoto 0:0e0631af0305 94 #endif //OPENCV_FLANN_GROUND_TRUTH_H_