1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
|
/*
Actionaz
Copyright (C) 2008-2014 Jonathan Mercier-Ganady
Actionaz is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Actionaz is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
Contact : jmgr@jmgr.info
*/
/***************************************************************************
* FastMatchTemplate.h
*
*
* Copyright 2010 Tristen Georgiou
* tristen_georgiou@hotmail.com
****************************************************************************/
/*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
How fast match template works:
1. Both target and source image are down sampled numDownPyrs times
2. cvMatchTemplate() function is called on shrunken images
(uses CCORR_NORMED algorithm)
3. The numMaxima best locations are found
4. For each point where a maxima was located:
a) Original source image is searched at point +/- searchExpansion
pixels in both x and y direction
5. If match score is above matchPercentage then the location and score is
saved in the foundPointsList and confidencesList, respectively
6. If findMultipleTargets is true, an attempt will be made to find up to
numMaxima targets
7. (Optional) The targets can be drawn to a color version of the source
image using the DrawFoundTargets() function
*/
/*
Adapted to the Actionaz project by Jonathan Mercier-Ganady
Original files: http://opencv.willowgarage.com/wiki/FastMatchTemplate
*/
#ifndef OPENCVALGORITHMS_H
#define OPENCVALGORITHMS_H
#include "actiontools_global.h"
#include "matchingpointlist.h"
#include <QObject>
#include <QList>
#include <QImage>
#include <QVector>
#include <QFuture>
#include <QFutureWatcher>
#include <QMetaType>
namespace cv
{
class Mat;
}
namespace ActionTools
{
class ACTIONTOOLSSHARED_EXPORT OpenCVAlgorithms : public QObject
{
Q_OBJECT
public:
enum AlgorithmError
{
NoError,
AlreadyRunningError,
SourceImageSmallerThanTargerImageError,
NotSameDepthError,
NotSameChannelCountError,
OpenCVException
};
enum AlgorithmMethod
{
CorrelationCoefficientMethod,
CrossCorrelationMethod,
SquaredDifferenceMethod
};
explicit OpenCVAlgorithms(QObject *parent = 0);
bool findSubImageAsync(const QList<QImage> &sources,
const QImage &target,
int matchPercentage = 70,
int maximumMatches = 10,
int downPyrs = 2,
int searchExpansion = 15,
AlgorithmMethod method = CorrelationCoefficientMethod);
bool findSubImage(const QList<QImage> &sources,
const QImage &target,
MatchingPointList &matchingPoints,
int matchPercentage = 70,
int maximumMatches = 10,
int downPyrs = 2,
int searchExpansion = 15,
AlgorithmMethod method = CorrelationCoefficientMethod);
void cancelSearch();
AlgorithmError error() const { return mError; }
const QString &errorString() const { return mErrorString; }
signals:
void finished(const ActionTools::MatchingPointList &matchingPointList);
private slots:
void finished();
private:
bool checkInputImages(const QList<cv::Mat> &sources, const cv::Mat &target);
/*=============================================================================
FastMatchTemplate
Performs a fast match template
Returns: true on success, false on failure
Parameters:
source - source image (where we are searching)
target - target image (what we are searching for)
foundPointsList - contains a list of the points where the target was found
confidencesList - contains a list of the confidence value (0-100) for each
found target
matchPercentage - the minimum required match score to consider the target
found
numMaxima - the maximum number of search locations to try before exiting
(i.e. when image is down-sampled and searched, we collect the
best numMaxima locations - those with the highest confidence -
and search the original image at these locations)
numDownPyrs - the number of times to down-sample the image (only increase
this number if your images are really large)
searchExpansion - The original source image is searched at the top locations
with +/- searchExpansion pixels in both the x and y
directions
*/
MatchingPointList fastMatchTemplate(const QList<cv::Mat> &sources,
const cv::Mat &target,
int matchPercentage,
int maximumMatches,
int downPyrs,
int searchExpansion,
AlgorithmMethod method);
static QVector<QPoint> multipleMinMaxLoc(const cv::Mat &image, int maximumMatches, AlgorithmMethod method);
static QImage toQImage(const cv::Mat &image);
static cv::Mat toCVMat(const QImage &image);
static int toOpenCVMethod(AlgorithmMethod method);
AlgorithmError mError;
QString mErrorString;
QFuture<MatchingPointList> mFuture;
QFutureWatcher<MatchingPointList> mFutureWatcher;
};
}
Q_DECLARE_METATYPE(ActionTools::MatchingPointList)
#endif // OPENCVALGORITHMS_H
|