biagio montesano
11 years ago
parent
3b207a77f8
commit
d191729b46
15 changed files with 1081 additions and 108 deletions
|
After Width: | Height: | Size: 23 KiB |
|
After Width: | Height: | Size: 26 KiB |
|
After Width: | Height: | Size: 792 KiB |
|
After Width: | Height: | Size: 51 KiB |
|
After Width: | Height: | Size: 56 KiB |
@ -0,0 +1,169 @@ |
||||
#include <opencv2/line_descriptor.hpp> |
||||
|
||||
#include "opencv2/core/utility.hpp" |
||||
#include "opencv2/core/private.hpp" |
||||
#include <opencv2/imgproc.hpp> |
||||
#include <opencv2/features2d.hpp> |
||||
#include <opencv2/highgui.hpp> |
||||
|
||||
#include <iostream> |
||||
#include <vector> |
||||
|
||||
using namespace cv; |
||||
|
||||
static const char* keys = |
||||
{ |
||||
"{@image_path1 | | Image path 1 }" |
||||
"{@image_path2 | | Image path 2 }" |
||||
}; |
||||
|
||||
static void help() |
||||
{ |
||||
std::cout << "\nThis example shows the functionalities of descriptors matching\n" << |
||||
"Please, run this sample using a command in the form\n" << |
||||
"./example_line_descriptor_matching <path_to_input_image 1>" |
||||
<< "<path_to_input_image 2>" << std::endl; |
||||
|
||||
} |
||||
|
||||
/* invert numBits bits in input char */ |
||||
uchar invertSingleBits (uchar dividend_char, int numBits) |
||||
{ |
||||
std::vector<int> bin_vector; |
||||
long dividend; |
||||
long bin_num; |
||||
|
||||
/* convert input char to a long */ |
||||
dividend = (long)dividend_char; |
||||
|
||||
/*if a 0 has been obtained, just generate a 8-bit long vector of zeros */ |
||||
if(dividend == 0) |
||||
bin_vector = std::vector<int>(8, 0); |
||||
|
||||
/* else, apply classic decimal to binary conversion */ |
||||
else |
||||
{ |
||||
while ( dividend >= 1 ) |
||||
{ |
||||
bin_num = dividend % 2; |
||||
dividend /= 2; |
||||
bin_vector.push_back(bin_num); |
||||
} |
||||
} |
||||
|
||||
/* ensure that binary vector always has length 8 */ |
||||
if(bin_vector.size()<8){ |
||||
std::vector<int> zeros (8-bin_vector.size(), 0); |
||||
bin_vector.insert(bin_vector.end(), zeros.begin(), zeros.end()); |
||||
} |
||||
|
||||
/* invert numBits bits */ |
||||
for(int index = 0; index<numBits; index++) |
||||
{ |
||||
if(bin_vector[index] == 0) |
||||
bin_vector[index] = 1; |
||||
|
||||
else |
||||
bin_vector[index] = 0; |
||||
} |
||||
|
||||
/* reconvert to decimal */ |
||||
uchar result; |
||||
for(int i = (int)bin_vector.size()-1; i>=0; i--) |
||||
result += bin_vector[i]*pow(2, i); |
||||
|
||||
return result; |
||||
} |
||||
|
||||
int main( int argc, char** argv ) |
||||
{ |
||||
/* get parameters from comand line */ |
||||
CommandLineParser parser( argc, argv, keys ); |
||||
String image_path1 = parser.get<String>( 0 ); |
||||
String image_path2 = parser.get<String>( 1 ); |
||||
|
||||
if(image_path1.empty() || image_path2.empty()) |
||||
{ |
||||
help(); |
||||
return -1; |
||||
} |
||||
|
||||
|
||||
/* load image */ |
||||
cv::Mat imageMat1 = imread(image_path1, 1); |
||||
cv::Mat imageMat2 = imread(image_path2, 1); |
||||
|
||||
if(imageMat1.data == NULL || imageMat2.data == NULL) |
||||
{ |
||||
std::cout << "Error, images could not be loaded. Please, check their paths" |
||||
<< std::endl; |
||||
} |
||||
|
||||
/* create binary masks */ |
||||
cv::Mat mask1 = Mat::ones(imageMat1.size(), CV_8UC1); |
||||
cv::Mat mask2 = Mat::ones(imageMat2.size(), CV_8UC1); |
||||
|
||||
/* create a pointer to a BinaryDescriptor object with default parameters */ |
||||
Ptr<BinaryDescriptor> bd = BinaryDescriptor::createBinaryDescriptor(); |
||||
|
||||
/* compute lines */ |
||||
std::vector<KeyLine> keylines1, keylines2; |
||||
bd->detect(imageMat1, keylines1, mask1); |
||||
bd->detect(imageMat2, keylines2, mask2); |
||||
|
||||
/* compute descriptors */ |
||||
cv::Mat descr1, descr2; |
||||
bd->compute(imageMat1, keylines1, descr1); |
||||
bd->compute(imageMat2, keylines2, descr2); |
||||
|
||||
/* create a BinaryDescriptorMatcher object */ |
||||
Ptr<BinaryDescriptorMatcher> bdm = BinaryDescriptorMatcher::createBinaryDescriptorMatcher(); |
||||
|
||||
/* make a copy of descr2 mat */ |
||||
Mat descr2Copy = descr1.clone(); |
||||
|
||||
/* randomly change some bits in original descriptors */ |
||||
srand (time(NULL)); |
||||
|
||||
for(int j = 0; j<descr1.rows; j++) |
||||
{ |
||||
/* select a random column */ |
||||
int randCol = rand() % 32; |
||||
|
||||
/* get correspondent data */ |
||||
uchar u = descr1.at<uchar>(j, randCol); |
||||
|
||||
/* change bits */ |
||||
for(int k = 1; k<=5; k++) |
||||
{ |
||||
/* copy current row to train matrix */ |
||||
descr2Copy.push_back(descr1.row(j)); |
||||
|
||||
/* invert k bits */ |
||||
uchar uc = invertSingleBits(u, k); |
||||
|
||||
/* update current row in train matrix */ |
||||
descr2Copy.at<uchar>(descr2Copy.rows-1, randCol) = uc; |
||||
} |
||||
} |
||||
|
||||
/* prepare a structure to host matches */ |
||||
std::vector<std::vector<DMatch> > matches; |
||||
|
||||
/* require knn match */ |
||||
bdm->knnMatch(descr1, descr2, matches, 6); |
||||
|
||||
/* visualize matches and Hamming distances */ |
||||
for(size_t v = 0; v<matches.size(); v++) |
||||
{ |
||||
for(size_t m = 0; m<matches[v].size(); m++) |
||||
{ |
||||
DMatch dm = matches[v][m]; |
||||
std::cout << dm.queryIdx << " " << dm.trainIdx << " " |
||||
<< dm.distance << std::endl; |
||||
} |
||||
} |
||||
|
||||
} |
||||
|
||||
|
||||
@ -0,0 +1,111 @@ |
||||
#include <opencv2/line_descriptor.hpp> |
||||
|
||||
#include "opencv2/core/utility.hpp" |
||||
#include "opencv2/core/private.hpp" |
||||
#include <opencv2/imgproc.hpp> |
||||
#include <opencv2/features2d.hpp> |
||||
#include <opencv2/highgui.hpp> |
||||
|
||||
#include <iostream> |
||||
#include <vector> |
||||
|
||||
using namespace cv; |
||||
|
||||
static const std::string images[] = |
||||
{ |
||||
"cameraman.jpg", |
||||
"church.jpg", |
||||
"church2.png", |
||||
"einstein.jpg", |
||||
"stuff.jpg" |
||||
}; |
||||
|
||||
static const char* keys = |
||||
{ |
||||
"{@image_path | | Image path }" |
||||
}; |
||||
|
||||
static void help() |
||||
{ |
||||
std::cout << "\nThis example shows the functionalities of radius matching " << |
||||
"Please, run this sample using a command in the form\n" << |
||||
"./example_line_descriptor_radius_matching <path_to_input_images>/" |
||||
<< std::endl; |
||||
} |
||||
|
||||
int main( int argc, char** argv ) |
||||
{ |
||||
/* get parameters from comand line */ |
||||
CommandLineParser parser( argc, argv, keys ); |
||||
String pathToImages = parser.get<String>( 0 ); |
||||
|
||||
/* create structures for hosting KeyLines and descriptors */ |
||||
int num_elements = sizeof( images ) / sizeof( images[0] ); |
||||
std::vector<Mat> descriptorsMat; |
||||
std::vector<std::vector<KeyLine> > linesMat; |
||||
|
||||
/*create a pointer to a BinaryDescriptor object */ |
||||
Ptr<BinaryDescriptor> bd = BinaryDescriptor::createBinaryDescriptor(); |
||||
|
||||
/* compute lines and descriptors */ |
||||
for(int i = 0; i<num_elements; i++) |
||||
{ |
||||
/* get path to image */ |
||||
std::stringstream image_path; |
||||
image_path << pathToImages << images[i]; |
||||
|
||||
/* load image */ |
||||
Mat loadedImage = imread(image_path.str().c_str(), 1); |
||||
if(loadedImage.data == NULL) |
||||
{ |
||||
std::cout << "Could not load images." << std::endl; |
||||
help(); |
||||
exit(-1); |
||||
} |
||||
|
||||
/* compute lines and descriptors */ |
||||
std::vector<KeyLine> lines; |
||||
Mat computedDescr; |
||||
bd->detect(loadedImage, lines); |
||||
bd->compute(loadedImage, lines, computedDescr); |
||||
|
||||
descriptorsMat.push_back(computedDescr); |
||||
linesMat.push_back(lines); |
||||
|
||||
} |
||||
|
||||
/* compose a queries matrix */ |
||||
Mat queries; |
||||
for(size_t j = 0; j<descriptorsMat.size(); j++) |
||||
{ |
||||
if(descriptorsMat[j].rows >= 5) |
||||
queries.push_back(descriptorsMat[j].rowRange(0, 5)); |
||||
|
||||
else if(descriptorsMat[j].rows >0 && descriptorsMat[j].rows<5) |
||||
queries.push_back(descriptorsMat[j]); |
||||
} |
||||
|
||||
std::cout << "It has been generated a matrix of " << queries.rows |
||||
<< " descriptors" << std::endl; |
||||
|
||||
/* create a BinaryDescriptorMatcher object */ |
||||
Ptr<BinaryDescriptorMatcher> bdm = BinaryDescriptorMatcher::createBinaryDescriptorMatcher(); |
||||
|
||||
/* populate matcher */ |
||||
bdm->add(descriptorsMat); |
||||
|
||||
/* compute matches */ |
||||
std::vector<std::vector<DMatch> > matches; |
||||
bdm->radiusMatch(queries, matches, 30); |
||||
|
||||
/* print matches */ |
||||
for(size_t q = 0; q<matches.size(); q++) |
||||
{ |
||||
for(size_t m = 0; m<matches[q].size(); m++) |
||||
{ |
||||
DMatch dm = matches[q][m]; |
||||
std::cout << "Descriptor: " << q << " Image: " << dm.imgIdx |
||||
<< " Distance: " << dm.distance << std::endl; |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,142 @@ |
||||
#include "precomp.hpp" |
||||
|
||||
namespace cv |
||||
{ |
||||
/* draw matches between two images */ |
||||
void drawLineMatches( const Mat& img1, const std::vector<KeyLine>& keylines1, |
||||
const Mat& img2, const std::vector<KeyLine>& keylines2, |
||||
const std::vector<DMatch>& matches1to2, |
||||
Mat& outImg, const Scalar& matchColor, |
||||
const Scalar& singleLineColor, |
||||
const std::vector<char>& matchesMask, int flags ) |
||||
{ |
||||
|
||||
/* initialize output matrix (if necessary) */ |
||||
if(flags == DrawLinesMatchesFlags::DEFAULT) |
||||
{ |
||||
/* check how many rows are necessary for output matrix */ |
||||
int totalRows = img1.rows >= img2.rows?img1.rows:img2.rows; |
||||
|
||||
/* initialize output matrix */ |
||||
outImg = Mat::zeros(totalRows, img1.cols+img2.cols, img1.type()); |
||||
|
||||
} |
||||
|
||||
/* initialize random seed: */ |
||||
srand (time(NULL)); |
||||
|
||||
Scalar singleLineColorRGB; |
||||
if(singleLineColor == Scalar::all(-1)) |
||||
{ |
||||
int R = (rand() % (int)(255 + 1)); |
||||
int G = (rand() % (int)(255 + 1)); |
||||
int B = (rand() % (int)(255 + 1)); |
||||
|
||||
singleLineColorRGB = Scalar(R, G, B); |
||||
} |
||||
|
||||
else |
||||
singleLineColorRGB = singleLineColor; |
||||
|
||||
/* copy input images to output images */ |
||||
Mat roi_left(outImg, Rect(0,0,img1.cols,img1.rows)); |
||||
Mat roi_right(outImg, Rect(img1.cols,0,img2.cols,img2.rows)); |
||||
img1.copyTo(roi_left); |
||||
img2.copyTo(roi_right); |
||||
|
||||
/* get columns offset */ |
||||
int offset = img1.cols; |
||||
|
||||
/* if requested, draw lines from both images */ |
||||
if(flags != DrawLinesMatchesFlags::NOT_DRAW_SINGLE_LINES) |
||||
{ |
||||
for(size_t i = 0; i<keylines1.size(); i++) |
||||
{ |
||||
KeyLine k1 = keylines1[i]; |
||||
line(outImg,Point(k1.startPointX, k1.startPointY), |
||||
Point(k1.endPointX, k1.endPointY), singleLineColorRGB, 2); |
||||
} |
||||
|
||||
for(size_t j = 0; j<keylines2.size(); j++) |
||||
{ |
||||
KeyLine k2 = keylines2[j]; |
||||
line(outImg,Point(k2.startPointX+offset, k2.startPointY), |
||||
Point(k2.endPointX+offset, k2.endPointY), singleLineColorRGB, 2); |
||||
} |
||||
} |
||||
|
||||
/* draw matches */ |
||||
for(size_t counter = 0; counter<matches1to2.size(); counter++) |
||||
{ |
||||
if(matchesMask[counter] != 0) |
||||
{ |
||||
DMatch dm = matches1to2[counter]; |
||||
KeyLine left = keylines1[dm.queryIdx]; |
||||
KeyLine right = keylines2[dm.trainIdx]; |
||||
|
||||
Scalar matchColorRGB; |
||||
if(matchColor == Scalar::all(-1)) |
||||
{ |
||||
int R = (rand() % (int)(255 + 1)); |
||||
int G = (rand() % (int)(255 + 1)); |
||||
int B = (rand() % (int)(255 + 1)); |
||||
|
||||
matchColorRGB = Scalar(R, G, B); |
||||
|
||||
if(singleLineColor == Scalar::all(-1)) |
||||
singleLineColorRGB = matchColorRGB; |
||||
} |
||||
|
||||
else |
||||
matchColorRGB = matchColor; |
||||
|
||||
/* draw lines if necessary */ |
||||
line(outImg, Point(left.startPointX, left.startPointY), |
||||
Point(left.endPointX, left.endPointY), singleLineColorRGB, 2); |
||||
|
||||
line(outImg, Point(right.startPointX+offset, right.startPointY), |
||||
Point(right.endPointX+offset, right.endPointY), singleLineColorRGB, 2); |
||||
|
||||
/* link correspondent lines */ |
||||
line(outImg, Point(left.startPointX, left.startPointY), |
||||
Point(right.startPointX+offset, right.startPointY), matchColorRGB, 1); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/* draw extracted lines on original image */ |
||||
void drawKeylines( const Mat& image, |
||||
const std::vector<KeyLine>& keylines, |
||||
Mat& outImage, |
||||
const Scalar& color, |
||||
int flags ) |
||||
{ |
||||
if(flags == DrawLinesMatchesFlags::DEFAULT) |
||||
outImage = image.clone(); |
||||
|
||||
for(size_t i = 0; i<keylines.size(); i++) |
||||
{ |
||||
/* decide lines' color */ |
||||
Scalar lineColor; |
||||
if(color != Scalar::all(-1)) |
||||
{ |
||||
int R = (rand() % (int)(255 + 1)); |
||||
int G = (rand() % (int)(255 + 1)); |
||||
int B = (rand() % (int)(255 + 1)); |
||||
|
||||
lineColor = Scalar(R, G, B); |
||||
} |
||||
|
||||
else |
||||
lineColor = color; |
||||
|
||||
/* get line */ |
||||
KeyLine k = keylines[i]; |
||||
|
||||
/* draw line */ |
||||
line(outImage, Point(k.startPointX, k.startPointY), |
||||
Point(k.endPointX, k.endPointY), lineColor, 1); |
||||
} |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue