407 lines
19 KiB
ReStructuredText
407 lines
19 KiB
ReStructuredText
Feature Detection and Description
|
|
=================================
|
|
|
|
.. highlight:: cpp
|
|
|
|
|
|
|
|
gpu::SURF_GPU
|
|
-------------
|
|
.. ocv:class:: gpu::SURF_GPU
|
|
|
|
Class used for extracting Speeded Up Robust Features (SURF) from an image. ::
|
|
|
|
class SURF_GPU : public CvSURFParams
|
|
{
|
|
public:
|
|
enum KeypointLayout
|
|
{
|
|
SF_X = 0,
|
|
SF_Y,
|
|
SF_LAPLACIAN,
|
|
SF_SIZE,
|
|
SF_DIR,
|
|
SF_HESSIAN,
|
|
SF_FEATURE_STRIDE
|
|
};
|
|
|
|
//! the default constructor
|
|
SURF_GPU();
|
|
//! the full constructor taking all the necessary parameters
|
|
explicit SURF_GPU(double _hessianThreshold, int _nOctaves=4,
|
|
int _nOctaveLayers=2, bool _extended=false, float _keypointsRatio=0.01f);
|
|
|
|
//! returns the descriptor size in float's (64 or 128)
|
|
int descriptorSize() const;
|
|
|
|
//! upload host keypoints to device memory
|
|
void uploadKeypoints(const vector<KeyPoint>& keypoints,
|
|
GpuMat& keypointsGPU);
|
|
//! download keypoints from device to host memory
|
|
void downloadKeypoints(const GpuMat& keypointsGPU,
|
|
vector<KeyPoint>& keypoints);
|
|
|
|
//! download descriptors from device to host memory
|
|
void downloadDescriptors(const GpuMat& descriptorsGPU,
|
|
vector<float>& descriptors);
|
|
|
|
void operator()(const GpuMat& img, const GpuMat& mask,
|
|
GpuMat& keypoints);
|
|
|
|
void operator()(const GpuMat& img, const GpuMat& mask,
|
|
GpuMat& keypoints, GpuMat& descriptors,
|
|
bool useProvidedKeypoints = false,
|
|
bool calcOrientation = true);
|
|
|
|
void operator()(const GpuMat& img, const GpuMat& mask,
|
|
std::vector<KeyPoint>& keypoints);
|
|
|
|
void operator()(const GpuMat& img, const GpuMat& mask,
|
|
std::vector<KeyPoint>& keypoints, GpuMat& descriptors,
|
|
bool useProvidedKeypoints = false,
|
|
bool calcOrientation = true);
|
|
|
|
void operator()(const GpuMat& img, const GpuMat& mask,
|
|
std::vector<KeyPoint>& keypoints,
|
|
std::vector<float>& descriptors,
|
|
bool useProvidedKeypoints = false,
|
|
bool calcOrientation = true);
|
|
|
|
void releaseMemory();
|
|
|
|
//! max keypoints = keypointsRatio * img.size().area()
|
|
float keypointsRatio;
|
|
|
|
GpuMat sum, mask1, maskSum, intBuffer;
|
|
|
|
GpuMat det, trace;
|
|
|
|
GpuMat maxPosBuffer;
|
|
};
|
|
|
|
|
|
The class ``SURF_GPU`` implements Speeded Up Robust Features descriptor. There is a fast multi-scale Hessian keypoint detector that can be used to find the keypoints (which is the default option). But the descriptors can also be computed for the user-specified keypoints. Only 8-bit grayscale images are supported.
|
|
|
|
The class ``SURF_GPU`` can store results in the GPU and CPU memory. It provides functions to convert results between CPU and GPU version ( ``uploadKeypoints``, ``downloadKeypoints``, ``downloadDescriptors`` ). The format of CPU results is the same as ``SURF`` results. GPU results are stored in ``GpuMat``. The ``keypoints`` matrix is :math:`\texttt{nFeatures} \times 6` matrix with the ``CV_32FC1`` type.
|
|
|
|
* ``keypoints.ptr<float>(SF_X)[i]`` contains x coordinate of the i-th feature.
|
|
* ``keypoints.ptr<float>(SF_Y)[i]`` contains y coordinate of the i-th feature.
|
|
* ``keypoints.ptr<float>(SF_LAPLACIAN)[i]`` contains the laplacian sign of the i-th feature.
|
|
* ``keypoints.ptr<float>(SF_SIZE)[i]`` contains the size of the i-th feature.
|
|
* ``keypoints.ptr<float>(SF_DIR)[i]`` contain orientation of the i-th feature.
|
|
* ``keypoints.ptr<float>(SF_HESSIAN)[i]`` contains the response of the i-th feature.
|
|
|
|
The ``descriptors`` matrix is :math:`\texttt{nFeatures} \times \texttt{descriptorSize}` matrix with the ``CV_32FC1`` type.
|
|
|
|
The class ``SURF_GPU`` uses some buffers and provides access to it. All buffers can be safely released between function calls.
|
|
|
|
.. seealso:: :ocv:class:`SURF`
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU
|
|
--------------------------
|
|
.. ocv:class:: gpu::BruteForceMatcher_GPU
|
|
|
|
Brute-force descriptor matcher. For each descriptor in the first set, this matcher finds the closest descriptor in the second set by trying each one. This descriptor matcher supports masking permissible matches between descriptor sets. ::
|
|
|
|
template<class Distance>
|
|
class BruteForceMatcher_GPU
|
|
{
|
|
public:
|
|
// Add descriptors to train descriptor collection.
|
|
void add(const std::vector<GpuMat>& descCollection);
|
|
|
|
// Get train descriptors collection.
|
|
const std::vector<GpuMat>& getTrainDescriptors() const;
|
|
|
|
// Clear train descriptors collection.
|
|
void clear();
|
|
|
|
// Return true if there are no train descriptors in collection.
|
|
bool empty() const;
|
|
|
|
// Return true if the matcher supports mask in match methods.
|
|
bool isMaskSupported() const;
|
|
|
|
void matchSingle(const GpuMat& queryDescs, const GpuMat& trainDescs,
|
|
GpuMat& trainIdx, GpuMat& distance,
|
|
const GpuMat& mask = GpuMat(), Stream& stream = Stream::Null());
|
|
|
|
static void matchDownload(const GpuMat& trainIdx,
|
|
const GpuMat& distance, std::vector<DMatch>& matches);
|
|
static void matchConvert(const Mat& trainIdx,
|
|
const Mat& distance, std::vector<DMatch>& matches);
|
|
|
|
void match(const GpuMat& queryDescs, const GpuMat& trainDescs,
|
|
std::vector<DMatch>& matches, const GpuMat& mask = GpuMat());
|
|
|
|
void makeGpuCollection(GpuMat& trainCollection, GpuMat& maskCollection,
|
|
const vector<GpuMat>& masks = std::vector<GpuMat>());
|
|
|
|
void matchCollection(const GpuMat& queryDescs,
|
|
const GpuMat& trainCollection,
|
|
GpuMat& trainIdx, GpuMat& imgIdx, GpuMat& distance,
|
|
const GpuMat& maskCollection, Stream& stream = Stream::Null());
|
|
|
|
static void matchDownload(const GpuMat& trainIdx, GpuMat& imgIdx,
|
|
const GpuMat& distance, std::vector<DMatch>& matches);
|
|
static void matchConvert(const Mat& trainIdx, const Mat& imgIdx,
|
|
const Mat& distance, std::vector<DMatch>& matches);
|
|
|
|
void match(const GpuMat& queryDescs, std::vector<DMatch>& matches,
|
|
const std::vector<GpuMat>& masks = std::vector<GpuMat>());
|
|
|
|
void knnMatch(const GpuMat& queryDescs, const GpuMat& trainDescs,
|
|
GpuMat& trainIdx, GpuMat& distance, GpuMat& allDist, int k,
|
|
const GpuMat& mask = GpuMat(), Stream& stream = Stream::Null());
|
|
|
|
static void knnMatchDownload(const GpuMat& trainIdx,
|
|
const GpuMat& distance, std::vector< std::vector<DMatch> >& matches,
|
|
bool compactResult = false);
|
|
static void knnMatchConvert(const Mat& trainIdx,
|
|
const Mat& distance, std::vector< std::vector<DMatch> >& matches,
|
|
bool compactResult = false);
|
|
|
|
void knnMatch(const GpuMat& queryDescs, const GpuMat& trainDescs,
|
|
std::vector< std::vector<DMatch> >& matches, int k,
|
|
const GpuMat& mask = GpuMat(), bool compactResult = false);
|
|
|
|
void knnMatch(const GpuMat& queryDescs,
|
|
std::vector< std::vector<DMatch> >& matches, int knn,
|
|
const std::vector<GpuMat>& masks = std::vector<GpuMat>(),
|
|
bool compactResult = false );
|
|
|
|
void radiusMatch(const GpuMat& queryDescs, const GpuMat& trainDescs,
|
|
GpuMat& trainIdx, GpuMat& nMatches, GpuMat& distance,
|
|
float maxDistance, const GpuMat& mask = GpuMat(), Stream& stream = Stream::Null());
|
|
|
|
static void radiusMatchDownload(const GpuMat& trainIdx,
|
|
const GpuMat& nMatches, const GpuMat& distance,
|
|
std::vector< std::vector<DMatch> >& matches,
|
|
bool compactResult = false);
|
|
static void radiusMatchConvert(const Mat& trainIdx,
|
|
const Mat& nMatches, const Mat& distance,
|
|
std::vector< std::vector<DMatch> >& matches,
|
|
bool compactResult = false);
|
|
|
|
void radiusMatch(const GpuMat& queryDescs, const GpuMat& trainDescs,
|
|
std::vector< std::vector<DMatch> >& matches, float maxDistance,
|
|
const GpuMat& mask = GpuMat(), bool compactResult = false);
|
|
|
|
void radiusMatch(const GpuMat& queryDescs,
|
|
std::vector< std::vector<DMatch> >& matches, float maxDistance,
|
|
const std::vector<GpuMat>& masks = std::vector<GpuMat>(),
|
|
bool compactResult = false);
|
|
|
|
private:
|
|
std::vector<GpuMat> trainDescCollection;
|
|
};
|
|
|
|
|
|
The class ``BruteForceMatcher_GPU`` has an interface similar to the class :ocv:class:`DescriptorMatcher`. It has two groups of ``match`` methods: for matching descriptors of one image with another image or with an image set. Also, all functions have an alternative to save results either to the GPU memory or to the CPU memory. The ``Distance`` template parameter is kept for CPU/GPU interfaces similarity. ``BruteForceMatcher_GPU`` supports only the ``L1<float>``, ``L2<float>``, and ``Hamming`` distance types.
|
|
|
|
.. seealso:: :ocv:class:`DescriptorMatcher`, :ocv:class:`BruteForceMatcher`
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::match
|
|
-------------------------------------
|
|
Finds the best match for each descriptor from a query set with train descriptors.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::match(const GpuMat& queryDescs, const GpuMat& trainDescs, std::vector<DMatch>& matches, const GpuMat& mask = GpuMat())
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::match(const GpuMat& queryDescs, std::vector<DMatch>& matches, const std::vector<GpuMat>& masks = std::vector<GpuMat>())
|
|
|
|
.. seealso:: :ocv:func:`DescriptorMatcher::match`
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::matchSingle
|
|
-------------------------------------------
|
|
Finds the best match for each query descriptor.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::matchSingle(const GpuMat& queryDescs, const GpuMat& trainDescs, GpuMat& trainIdx, GpuMat& distance, const GpuMat& mask = GpuMat(), Stream& stream = Stream::Null())
|
|
|
|
:param queryDescs: Query set of descriptors.
|
|
|
|
:param trainDescs: Training set of descriptors. It is not added to train descriptors collection stored in the class object.
|
|
|
|
:param trainIdx: Output matrix that contains the best train index for each query.
|
|
|
|
:param distance: Output matrix that contains the best distance for each query.
|
|
|
|
:param mask: Mask specifying permissible matches between the input query and train matrices of descriptors.
|
|
|
|
:param stream: Stream for the asynchronous version.
|
|
|
|
Results are stored in the GPU memory.
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::matchCollection
|
|
-----------------------------------------------
|
|
Finds the best match for each query descriptor from train collection.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::matchCollection(const GpuMat& queryDescs, const GpuMat& trainCollection, GpuMat& trainIdx, GpuMat& imgIdx, GpuMat& distance, const GpuMat& maskCollection, Stream& stream = Stream::Null())
|
|
|
|
:param queryDescs: Query set of descriptors.
|
|
|
|
:param trainCollection: :ocv:class:`gpu::GpuMat` containing train collection. It can be obtained from the collection of train descriptors that was set using the ``add`` method by :ocv:func:`gpu::BruteForceMatcher_GPU::makeGpuCollection`. Or it may contain a user-defined collection. This is a one-row matrix where each element is ``DevMem2D`` pointing out to a matrix of train descriptors.
|
|
|
|
:param trainIdx: Output matrix that contains the best train index for each query.
|
|
|
|
:param imgIdx: Output matrix that contains image train index for each query.
|
|
|
|
:param distance: Output matrix that contains the best distance for each query.
|
|
|
|
:param maskCollection: ``GpuMat`` containing a set of masks. It can be obtained from ``std::vector<GpuMat>`` by :ocv:func:`gpu::BruteForceMatcher_GPU::makeGpuCollection` or it may contain a user-defined mask set. This is an empty matrix or one-row matrix where each element is a ``PtrStep`` that points to one mask.
|
|
|
|
:param stream: Stream for the asynchronous version.
|
|
|
|
Results are stored in the GPU memory.
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::makeGpuCollection
|
|
-------------------------------------------------
|
|
Performs a GPU collection of train descriptors and masks in a suitable format for the :ocv:func:`gpu::BruteForceMatcher_GPU::matchCollection` function.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::makeGpuCollection(GpuMat& trainCollection, GpuMat& maskCollection, const vector<GpuMat>&masks = std::vector<GpuMat>())
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::matchDownload
|
|
---------------------------------------------
|
|
Downloads matrices obtained via :ocv:func:`gpu::BruteForceMatcher_GPU::matchSingle` or :ocv:func:`gpu::BruteForceMatcher_GPU::matchCollection` to vector with :ocv:class:`DMatch`.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::matchDownload(const GpuMat& trainIdx, const GpuMat& distance, std::vector<DMatch>&matches)
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::matchDownload(const GpuMat& trainIdx, GpuMat& imgIdx, const GpuMat& distance, std::vector<DMatch>&matches)
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::matchConvert
|
|
---------------------------------------------
|
|
Converts matrices obtained via :ocv:func:`gpu::BruteForceMatcher_GPU::matchSingle` or :ocv:func:`gpu::BruteForceMatcher_GPU::matchCollection` to vector with :ocv:class:`DMatch`.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::matchConvert(const Mat& trainIdx, const Mat& distance, std::vector<DMatch>&matches)
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::matchConvert(const Mat& trainIdx, const Mat& imgIdx, const Mat& distance, std::vector<DMatch>&matches)
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::knnMatch
|
|
----------------------------------------
|
|
Finds the k best matches for each descriptor from a query set with train descriptors.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::knnMatch(const GpuMat& queryDescs, const GpuMat& trainDescs, std::vector< std::vector<DMatch> >&matches, int k, const GpuMat& mask = GpuMat(), bool compactResult = false)
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::knnMatch(const GpuMat& queryDescs, std::vector< std::vector<DMatch> >&matches, int k, const std::vector<GpuMat>&masks = std::vector<GpuMat>(), bool compactResult = false )
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::knnMatch(const GpuMat& queryDescs, const GpuMat& trainDescs, GpuMat& trainIdx, GpuMat& distance, GpuMat& allDist, int k, const GpuMat& mask = GpuMat(), Stream& stream = Stream::Null())
|
|
|
|
:param queryDescs: Query set of descriptors.
|
|
|
|
:param trainDescs: Training set of descriptors. It is not be added to train descriptors collection stored in the class object.
|
|
|
|
:param trainIdx: Output matrix that contains the best train index for each query.
|
|
|
|
:param distance: Output matrix that contains the best distance for each query.
|
|
|
|
:param allDist: Output matrix that contains all distances between each query descriptors and each train descriptor.
|
|
|
|
:param k: Number of the best matches per each query descriptor (or less if it is not possible).
|
|
|
|
:param mask: Mask specifying permissible matches between the input query and train matrices of descriptors.
|
|
|
|
:param compactResult: If ``compactResult`` is ``true`` , the ``matches`` vector does not contain matches for fully masked-out query descriptors.
|
|
|
|
:param stream: Stream for the asynchronous version.
|
|
|
|
The function returns detected k (or less if not possible) matches in the increasing order by distance.
|
|
|
|
The third variant of the method stores the results in GPU memory.
|
|
|
|
.. seealso:: :ocv:func:`DescriptorMatcher::knnMatch`
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::knnMatchDownload
|
|
------------------------------------------------
|
|
Downloads matrices obtained via :ocv:func:`gpu::BruteForceMatcher_GPU::knnMatch` to vector with :ocv:class:`DMatch`.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::knnMatchDownload(const GpuMat& trainIdx, const GpuMat& distance, std::vector< std::vector<DMatch> >&matches, bool compactResult = false)
|
|
|
|
If ``compactResult`` is ``true`` , the ``matches`` vector does not contain matches for fully masked-out query descriptors.
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::knnMatchConvert
|
|
------------------------------------------------
|
|
Converts matrices obtained via :ocv:func:`gpu::BruteForceMatcher_GPU::knnMatch` to CPU vector with :ocv:class:`DMatch`.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::knnMatchConvert(const Mat& trainIdx, const Mat& distance, std::vector< std::vector<DMatch> >&matches, bool compactResult = false)
|
|
|
|
If ``compactResult`` is ``true`` , the ``matches`` vector does not contain matches for fully masked-out query descriptors.
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::radiusMatch
|
|
-------------------------------------------
|
|
For each query descriptor, finds the best matches with a distance less than a given threshold.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::radiusMatch(const GpuMat& queryDescs, const GpuMat& trainDescs, std::vector< std::vector<DMatch> >&matches, float maxDistance, const GpuMat& mask = GpuMat(), bool compactResult = false)
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::radiusMatch(const GpuMat& queryDescs, std::vector< std::vector<DMatch> >&matches, float maxDistance, const std::vector<GpuMat>&masks = std::vector<GpuMat>(), bool compactResult = false)
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::radiusMatch(const GpuMat& queryDescs, const GpuMat& trainDescs, GpuMat& trainIdx, GpuMat& nMatches, GpuMat& distance, float maxDistance, const GpuMat& mask = GpuMat(), Stream& stream = Stream::Null())
|
|
|
|
:param queryDescs: Query set of descriptors.
|
|
|
|
:param trainDescs: Training set of descriptors. It is not added to train descriptors collection stored in the class object.
|
|
|
|
:param trainIdx: Output matrix that contains the best train index for each query.
|
|
|
|
:param nMatches: Output matrix that contains the number of matching descriptors for each query.
|
|
|
|
:param distance: Output matrix that contains the best distance for each query.
|
|
|
|
:param maxDistance: Distance threshold.
|
|
|
|
:param mask: Mask specifying permissible matches between the input query and train matrices of descriptors.
|
|
|
|
:param compactResult: If ``compactResult`` is ``true`` , the ``matches`` vector does not contain matches for fully masked-out query descriptors.
|
|
|
|
:param stream: Stream for the asynchronous version.
|
|
|
|
The function returns detected matches in the increasing order by distance.
|
|
|
|
The methods work only on devices with the compute capability :math:`>=` 1.1.
|
|
|
|
The third variant of the method stores the results in GPU memory and does not store the points by the distance.
|
|
|
|
.. seealso:: :ocv:func:`DescriptorMatcher::radiusMatch`
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::radiusMatchDownload
|
|
---------------------------------------------------
|
|
Downloads matrices obtained via :ocv:func:`gpu::BruteForceMatcher_GPU::radiusMatch` to vector with :ocv:class:`DMatch`.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::radiusMatchDownload(const GpuMat& trainIdx, const GpuMat& nMatches, const GpuMat& distance, std::vector< std::vector<DMatch> >&matches, bool compactResult = false)
|
|
|
|
If ``compactResult`` is ``true`` , the ``matches`` vector does not contain matches for fully masked-out query descriptors.
|
|
|
|
|
|
|
|
|
|
gpu::BruteForceMatcher_GPU::radiusMatchConvert
|
|
---------------------------------------------------
|
|
Converts matrices obtained via :ocv:func:`gpu::BruteForceMatcher_GPU::radiusMatch` to vector with :ocv:class:`DMatch`.
|
|
|
|
.. ocv:function:: void gpu::BruteForceMatcher_GPU::radiusMatchConvert(const Mat& trainIdx, const Mat& nMatches, const Mat& distance, std::vector< std::vector<DMatch> >&matches, bool compactResult = false)
|
|
|
|
If ``compactResult`` is ``true`` , the ``matches`` vector does not contain matches for fully masked-out query descriptors.
|
|
|