opencv 常用脚本合集

一.批量调整图片大小（c++）

在处理数据集的时候，有时候会需要把直接的数据集图片大小进行调整，所以写了个脚本

/*
调整图片大小
*/
#include "opencv2/imgproc.hpp"
#include "opencv2/highgui.hpp"
#include "opencv2/ml.hpp"
#include "opencv2/objdetect.hpp"
#include <string>
using namespace std;
using namespace cv;
vector<string> getOrderedNames(vector<cv::String> filenames, cv::String folder) {
	vector<string> nameArr;
	string  suffixName;
	for (size_t i = 0; i < filenames.size(); ++i)
	{
		string::size_type iPos = filenames[i].find_last_of('\\') + 1;
		string filename = filenames[i].substr(iPos, filenames[i].length() - iPos);
		string name = filename.substr(0, filename.rfind("."));
		suffixName = filename.substr(filename.rfind("."), filename.length() - 1);//获取后缀名
		nameArr.emplace_back(name);
	}
	sort(nameArr.begin(), nameArr.end(), [](string a, string b) {return stoi(a) < stoi(b); });

	vector<string> orderedFilenames;
	for (size_t i = 0; i < filenames.size(); ++i)
	{

		string orderedName = folder + "\\" + nameArr[i] + suffixName;
		orderedFilenames.emplace_back(orderedName);
		//cout << "改变后：" << orderedFilenames[i] << endl;
	}
	return orderedFilenames;
}


int main()
{
	 int imgs_h = 480;//想要的宽高
	 int imgs_w = 640;
	std::vector<cv::String> filenames; // notice here that we are using the Opencv's embedded "String" class
	cv::String folder = "C:\\Users\\大喵喵\\Desktop\\大气湍流\\turbulence_dataset\\algorithm_simulated_videos\\BSD\\train\\3\\Blur\\RGB"; // again we are using the Opencv's embedded "String" class

	cv::glob(folder, filenames); // new function that does the job ;-)
	size_t count = filenames.size();
	if (count == 0)
	{
		std::cout << "file " << folder << " not  exits" << std::endl;
		return -1;
	}

	vector<string> orderedFilenames = getOrderedNames(filenames, folder);

	for (size_t i = 0; i < filenames.size(); ++i)
	{
		cout << "顺序：" << orderedFilenames[i] << endl;
		cv::Mat srcImage = imread(orderedFilenames[i]);
		
		resize(srcImage, srcImage, Size(imgs_w, imgs_h), 0, 0, INTER_LINEAR);
		imwrite(orderedFilenames[i], srcImage);
		
	}

}

二.OpenCV常用图像拼接方法 ：直接拼接(硬拼)

来源：[1]

// 01_Combine_Two_Images.cpp 
// 环境 VS2017 + OpenCV4.4.0
// 功能介绍：用于将两张图片拼接成一张大图(以左右拼接为例)，俗称的硬拼方法
// 特点：简单粗暴，现实应用对拍摄条件要求苛刻，适用性差
//#include "pch.h"
#include <iostream>
#include <math.h>
#include <opencv2/opencv.hpp>
 
using namespace std;
using namespace cv;
 
int main()
{
	Mat img1 = imread("01.jpg");
	Mat img2 = imread("02.jpg");
	if (img1.empty() || img2.empty())
	{
		cout << "加载图片失败，请检查对应路径图片是否存在！" << endl;
		return 1;
	}
	imshow("src1", img1);
	imshow("src2", img2);
	int w1 = img1.cols; int h1 = img1.rows;
	int w2 = img2.cols; int h2 = img2.rows;
	int width = w1 + w2; int height = max(h1, h2);
	Mat  resultImg = Mat(height, width, CV_8UC3, Scalar::all(0));
	Mat ROI_1 = resultImg(Rect(0, 0, w1, h1));
	Mat ROI_2 = resultImg(Rect(w1, 0, w2, h2));
	img1.copyTo(ROI_1);
	img2.copyTo(ROI_2);
	imshow("result", resultImg);
	imwrite("result.jpg", resultImg);
	waitKey(0);
	return 0;
}