void test()
{
Mat originalImage;
originalImage = imread("/Users/rzf/Downloads/dogs/src.jpg");
//【1】图片集的采集与处理
int Images_number = 40;//图片集中图片的数量
int step_x = originalImage.cols / Images_number; //将图片剪裁为80*45大小
int step_y = originalImage.rows / Images_number;
Mat srcImage;
vector<Mat> load_Images;//图片集容器
char filename[Images_number];//存储图片名字
for (int i = 1; i < Images_number + 1; i++) {
Mat dstImage;
sprintf(filename, "/Users/rzf/Downloads/dogs/dog%d.jpg", i);
cout << filename << endl;
srcImage = imread(filename); //Mat类图像
resize(srcImage, dstImage, Size(step_x, step_y), 0, 0, INTER_NEAREST);
load_Images.push_back(dstImage);
}
cout << "【1】图片加载完毕" << endl;
// imshow("dog", originalImage);
// resize(originalImage, showImage, Size(1600, 2700));
cout << "【2】显示原图完毕" << endl;
int width = originalImage.cols; //模板图的长宽
int height = originalImage.rows;
Mat frame;
vector<MatND> hsit_list;//直方图容器
//计算直方图的参数准备
int bins = 128; //直条数
int hist_sizes[] = { bins,bins,bins }; //存放每个维度的直方图尺寸数组 // 均为256条宽度
float range[] = { 0,256 }; //每一维数组的取值范围 // 均为0-255高度
const float* ranges[] = { range,range,range };
int channels[] = { 0,1,2 };
//for循环计算图片集的直方图并存储
for (int i = 0; i < Images_number; i++) {
/*load_Images[i].copyTo(frame);*/
MatND hsit_RGB;
Mat frame;
load_Images[i].copyTo(frame);
calcHist(&frame, 1, channels, Mat(), hsit_RGB, 3, hist_sizes, ranges, true, false);
hsit_list.push_back(hsit_RGB);
}
cout << "【3】计算直方图完毕" << endl;
//【4】遍历,寻找最匹配的图片并替换
int number_order;
for (int x = 0; x < height; x = x + step_y) {
for (int y = 0; y < width; y = y + step_x) {
if (y + step_x > width || x + step_y > height)
{
continue;
}
Mat roiImage = originalImage(Rect(y, x, step_x, step_y));//Rect(y, x, step_x, step_y)
//参数准备
MatND hsit_roi;
double match_ = 0.0;//匹配度
calcHist(&roiImage, 1, channels, Mat(), hsit_roi, 3, hist_sizes, ranges, true, false);
for (int i = 0; i < Images_number; i++) {
double match;
match = compareHist(hsit_roi, hsit_list[i], HISTCMP_CORREL);
if (match > match_) {//寻找对比对最高的
number_order = i;
match_ = match;
}
}
load_Images[number_order].copyTo(roiImage);
}
}
cout << "【4】遍历,寻找最匹配的图片并替换成功" << endl;
imwrite("/Users/rzf/Downloads/dogs/src1.jpg", originalImage);
}
图片的数量一定要多,这样效果才会好,我的图片少效果没那么好。
原图
拼成的图片