double compare(const Mat &image)
{
input = hist.colorReduce(image,div);
inputH = hist.getHistogram(input);
// return compareHist(refH,inputH,CV_COMP_BHATTACHARYYA);
//return compareHist(refH,inputH,CV_COMP_CHISQR);
//return compareHist(refH,inputH,CV_COMP_INTERSECT);
return compareHist(refH,inputH,CV_COMP_CORREL);
}
int main()
{
// Read input image
cv::Mat image= cv::imread("waves.jpg",0);
if (!image.data)
return 0;
// define image ROI
cv::Mat imageROI;
imageROI= image(cv::Rect(216,33,24,30)); // Cloud region
// Display reference patch
cv::namedWindow("Reference");
cv::imshow("Reference",imageROI);
// Find histogram of reference
Histogram1D h;
cv::Mat hist= h.getHistogram(imageROI);
cv::namedWindow("Reference Hist");
cv::imshow("Reference Hist",h.getHistogramImage(imageROI));
// Create the content finder
ContentFinder finder;
// set histogram to be back-projected
finder.setHistogram(hist);
finder.setThreshold(-1.0f);
// Get back-projection
cv::Mat result1;
result1= finder.find(image);
// Create negative image and display result
cv::Mat tmp;
result1.convertTo(tmp,CV_8U,-1.0,255.0);
cv::namedWindow("Backprojection result");
cv::imshow("Backprojection result",tmp);
// Get binary back-projection
finder.setThreshold(0.12f);
result1= finder.find(image);
// Draw a rectangle around the reference area
cv::rectangle(image, cv::Rect(216, 33, 24, 30), cv::Scalar(0, 0, 0));
// Display image
cv::namedWindow("Image");
cv::imshow("Image",image);
// Display result
cv::namedWindow("Detection Result");
cv::imshow("Detection Result",result1);
// Load color image
ColorHistogram hc;
cv::Mat color= cv::imread("waves.jpg");
// extract region of interest
imageROI= color(cv::Rect(0,0,100,45)); // blue sky area
// Get 3D colour histogram (8 bins per channel)
hc.setSize(8); // 8x8x8
cv::Mat shist= hc.getHistogram(imageROI);
// set histogram to be back-projected
finder.setHistogram(shist);
finder.setThreshold(0.05f);
// Get back-projection of color histogram
result1= finder.find(color);
cv::namedWindow("Color Detection Result");
cv::imshow("Color Detection Result",result1);
// Second color image
cv::Mat color2= cv::imread("dog.jpg");
cv::namedWindow("Second Image");
cv::imshow("Second Image",color2);
// Get back-projection of color histogram
cv::Mat result2= finder.find(color2);
cv::namedWindow("Result color (2)");
cv::imshow("Result color (2)",result2);
// Get ab color histogram
hc.setSize(256); // 256x256
cv::Mat colorhist= hc.getabHistogram(imageROI);
// display 2D histogram
colorhist.convertTo(tmp,CV_8U,-1.0,255.0);
cv::namedWindow("ab histogram");
cv::imshow("ab histogram",tmp);
// set histogram to be back-projected
finder.setHistogram(colorhist);
finder.setThreshold(0.05f);
// Convert to Lab space
cv::Mat lab;
cv::cvtColor(color, lab, CV_BGR2Lab);
// Get back-projection of ab histogram
int ch[2]={1,2};
result1= finder.find(lab,0,256.0f,ch);
cv::namedWindow("Result ab (1)");
cv::imshow("Result ab (1)",result1);
// Second colour image
cv::cvtColor(color2, lab, CV_BGR2Lab);
// Get back-projection of ab histogram
result2= finder.find(lab,0,256.0,ch);
cv::namedWindow("Result ab (2)");
cv::imshow("Result ab (2)",result2);
// Draw a rectangle around the reference sky area
cv::rectangle(color,cv::Rect(0,0,100,45),cv::Scalar(0,0,0));
cv::namedWindow("Color Image");
cv::imshow("Color Image",color);
// Get Hue colour histogram
hc.setSize(180); // 180 bins
colorhist= hc.getHueHistogram(imageROI);
// set histogram to be back-projected
finder.setHistogram(colorhist);
// Convert to HSV space
cv::Mat hsv;
cv::cvtColor(color, hsv, CV_BGR2HSV);
// Get back-projection of hue histogram
ch[0]=0;
result1= finder.find(hsv,0.0f,180.0f,ch);
cv::namedWindow("Result Hue (1)");
cv::imshow("Result Hue (1)",result1);
// Second colour image
color2= cv::imread("dog.jpg");
// Convert to HSV space
cv::cvtColor(color2, hsv, CV_BGR2HSV);
// Get back-projection of hue histogram
result2= finder.find(hsv,0.0f,180.0f,ch);
cv::namedWindow("Result Hue (2)");
cv::imshow("Result Hue (2)",result2);
cv::waitKey();
return 0;
}
void setRefrenceImage(const Mat &image)
{
reference = hist.colorReduce(image,div);
refH = hist.getHistogram(reference);
}