generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added handling of ROI in stitching features matchers

Browse Source
This commit is contained in:
Alexey Spizhevoy
2011-09-26 09:59:13 +00:00
parent b053a3b486
commit a7d053f10d
4 changed files with 51 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
modules/stitching/src/matchers.cpp
View File

@@ -249,10 +249,47 @@ namespace detail {
void FeaturesFinder::operator ()(const Mat &image, ImageFeatures &features)
{
find(image, features);
cout << features.descriptors.cols << " " << features.descriptors.rows << " " << features.descriptors.type() << endl;
features.img_size = image.size();
}
// TODO add tests for this function
void FeaturesFinder::operator ()(const Mat &image, ImageFeatures &features, const vector<Rect> &rois)
{
vector<ImageFeatures> roi_features;
size_t total_kps_count = 0;
int total_descriptors_width = 0;
for (size_t i = 0; i < rois.size(); ++i)
{
find(image(rois[i]), roi_features[i]);
total_kps_count += roi_features[i].keypoints.size();
total_descriptors_width += roi_features[i].descriptors.cols;
}
features.img_size = image.size();
features.keypoints.resize(total_kps_count);
features.descriptors.create(1, total_descriptors_width, roi_features[0].descriptors.type());
int kp_idx = 0;
int descr_offset = 0;
for (size_t i = 0; i < rois.size(); ++i)
{
for (size_t j = 0; j < features.keypoints.size(); ++j, ++kp_idx)
{
features.keypoints[kp_idx] = roi_features[i].keypoints[j];
features.keypoints[kp_idx].pt.x += (float)rois[i].x;
features.keypoints[kp_idx].pt.y += (float)rois[i].y;
}
Mat subdescr = features.descriptors.colRange(
descr_offset, descr_offset + roi_features[i].descriptors.cols);
roi_features[i].descriptors.copyTo(subdescr);
descr_offset += roi_features[i].descriptors.cols;
}
}
SurfFeaturesFinder::SurfFeaturesFinder(double hess_thresh, int num_octaves, int num_layers,
int num_octaves_descr, int num_layers_descr)
{

12
modules/stitching/src/stitcher.cpp
View File

@@ -100,6 +100,13 @@ Stitcher::Status Stitcher::stitch(InputArray imgs, OutputArray pano)
}
Stitcher::Status Stitcher::stitch(InputArray imgs, const vector<vector<Rect> > &rois, OutputArray pano)
{
rois_ = rois;
return stitch(imgs, pano);
}
Stitcher::Status Stitcher::matchImages()
{
if ((int)imgs_.size() < 2)
@@ -148,7 +155,10 @@ Stitcher::Status Stitcher::matchImages()
is_seam_scale_set = true;
}
(*features_finder_)(img, features_[i]);
if (rois_.empty())
(*features_finder_)(img, features_[i]);
else
(*features_finder_)(img, features_[i], rois_[i]);
features_[i].img_idx = i;
LOGLN("Features in image #" << i+1 << ": " << features_[i].keypoints.size());