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;
}
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。