程序功能:
将特定目录下的jpg与png图片文件批量重命名到另一目录下;
用途
图像标注样本收集中,从网络或实际拍摄收集的大量图片样本名称个不一致,在进行样本库标注录入时常常不是很方便,因此,可以采用将图片以有规律的方式批量重命名,以方便后续的标注与样本训练;
完整代码:
/***************************************************************************
程序功能:特定目录下的jpg与png图片文件批量重命名到另一目录下
****************************************************************************/
#include <sys/types.h>
#include <dirent.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define PATHNAME_LEN 80
#define PREFIX_LEN 16
#define SUFFIX_LEN 12
#define NEWNAME_LEN 30
#define FOUR_CHAR 4
// 截取特定位置的子串
char *strcut(const char *srcStr, int startPos, int cutLength);
int main(int argc, char *argv[])
{
if (argc < 3) {
printf("The order is: exe... srcDir... destDir...\n");
exit(EXIT_SUCCESS);
}
DIR *dir;
struct dirent *ptr;
int jpgCount = 0, pngCount = 0;
// prefixStr为用户编辑的文件批量命名前缀
char prefixStr[PREFIX_LEN] = {0};
printf("Enter the file prefix you want to modify here.\n");
scanf("%s", prefixStr);
// oldPath原路经,newPath目标存储路径
char *oldPath = argv[1];
char *newPath = argv[2];
dir = opendir((const char*)argv[1]);
while ((ptr = readdir(dir)) != NULL) // 遍历文件目录
{
if ((!strncmp(ptr->d_name,".",1)) || (!strncmp(ptr->d_name,"..",2))) { // 过滤掉目录中默认含有的.和..
continue;
}
// jpg与png文件区分命名
int cutStartPos = strlen(ptr->d_name) - FOUR_CHAR;
char *picSuffix = strcut(ptr->d_name, cutStartPos, FOUR_CHAR); // 原图片路径字符串中截取.jpg与.png后缀
char numSuffixName[SUFFIX_LEN] = {0}; // 数字后缀名
char suffixAllName[NEWNAME_LEN] = {0}; // 图片文件全名,不超过30个字符
char srcRoad[PATHNAME_LEN] = {0};
char destRoad[PATHNAME_LEN] = {0};
strcat(suffixAllName, prefixStr);
if (strcmp(picSuffix, ".jpg") == 0) {
sprintf(numSuffixName, "%06d.jpg", jpgCount);
strcat(suffixAllName, numSuffixName); // 拼接图片名与其数字后缀
strcat(strcat(srcRoad, oldPath), ptr->d_name);
strcat(strcat(destRoad, newPath), suffixAllName);
if (rename(srcRoad, destRoad) == -1) {
printf("error occur....\n");
exit(EXIT_FAILURE);
}
memset(srcRoad, '0', sizeof(srcRoad));
memset(destRoad, '0', sizeof(destRoad));
jpgCount++;
}
if (strcmp(picSuffix, ".png") == 0) {
sprintf(numSuffixName, "%06d.png", pngCount);
strcat(suffixAllName, numSuffixName);
strcat(strcat(srcRoad, oldPath), ptr->d_name);
strcat(strcat(destRoad, newPath), suffixAllName);
if (rename(srcRoad, destRoad) == -1) {
printf("error occur....\n");
exit(EXIT_FAILURE);
}
memset(srcRoad, '0', sizeof(srcRoad));
memset(destRoad, '0', sizeof(destRoad));
pngCount++;
}
}
printf("Copy completed......\n");
closedir(dir);
return 0;
}
char *strcut(const char *srcStr, int startPos, int cutLength)
{
if (startPos < 0 || cutLength < 0) {
return NULL;
}
int index = 0;
static char destArray[PATHNAME_LEN] = {0};
srcStr = srcStr + startPos;
while(srcStr && index < cutLength) {
destArray[index] = *srcStr++;
if (*srcStr == '\0') {
break;
}
index++;
}
return destArray;
}
运行演示截图
** 说明:测试图片源路径为/home/zyh/pic/ 目标路径为:/home/zyh/jpg/;初始时:源路径中存储着名称不规则的批量jpg与png图片,目标路径为空;运行后:源路径中jpg与png文件被移动至目标路径,目标路径中相关图片以特定规则名称展现,且jpg文件与png文件为分开命名;**
运行前
运行后
附:小白的代码仅实现最简单的图片文件批量按规律移动重命名,运行效率与代码精简度还有许多改进空间,欢迎各位大佬不吝赐教哈!