数据集的预处理工作量较大,为此,在Linux下借助OpenCV写了一个比较简单好用的图片批量修改尺寸和文件名的程序。
1、文件名的批量修改主要参考《UNIX环境高级编程》(第3版) 中ls命令的实现原理,用到opendir()、readdir()、closedir()、rename()函数和dirent结构体无需知道批量文件原文件名的前提下,实现文件的批量重命名。特别的,过滤到了 .和.. 文件夹。
2、图像尺寸的批量的修改是在文件名已知的前提下,利用OpenCV中的imread()、resize()、imwrite()函数实现。
注意事项: 函数定义:int rename(const char * oldpath,const char * newpath);因rename()会将参数oldpath 所指定的文件名称改为参数newpath所指的文件名称。若newpath所指定的文件已存在,则会被删除。所以为防止多次调用程序导致数据集部分内容丢失,oldpath和newpath不要为同一个文件夹(同一个文件夹,执行两遍程序,数据集被删除近一半)。注意:程序执行后,oldpath文件夹将为空。
#include<dirent.h>
#include<opencv2/opencv.hpp>
#include<string>
#include<errno.h>
#include <unistd.h>
using namespace std;
using namespace cv;
int main()
{
int i = 1;
DIR *dp;
char oldname[100];
char newname[100];
struct dirent *dirp;
const char* oldpath = "./data5/";
const char* newpath = "./data4/";
if((dp = opendir(oldpath)) == NULL)
printf("can't open file");
while((dirp = readdir(dp)) != NULL)
{
if(strcmp(dirp->d_name,".") && strcmp(dirp->d_name,".."))
{
//1、批量修改文件名
sprintf(newname, "%sn%d.jpg",newpath,i++);
sprintf(oldname, "%s%s",oldpath,dirp->d_name);
int temp = rename(oldname,newname); //将前者修改为后者
if(temp == -1) //如果修改文件出错
{
printf("修改文件名失败errno is: %d\n",errno); //打印出错原因
}
else
{
printf("文件名称修改成功\n");
//2、批量修改文件尺寸
Mat inputImg = imread(newname, 0);
Mat resizea;
resizea.create(32, 32, CV_8UC1);
resize(inputImg, resizea, resizea.size(), 0, 0, INTER_CUBIC);
imwrite(newname, resizea);
}
}
}
closedir(dp);
return 0;
}
Linux下编译命令:
g++ -o mui_img mui_img.cpp -lopencv_core -lopencv_highgui -lopencv_imgcodecs -lopencv_imgproc -lstdc++
欢迎学习交流,转载请注明出处https://blog.csdn.net/hzqgangtiexia/article/details/80385086