在制作数据集的过程中发现找到的图片名称和格式都很混乱,找了别人的发现老是报错.
同时总结了一下我自己这边程序报的错误:
1..jpg已存在**
比如我现在存在2.jpg,通过代码修改也会得到2.jpg,可是这并不是同一张图片,而且不能覆盖
–解决方法–
直接全选所有图片,重命名,然后随便取个名字,不要纯数字(纯数字也可以,不要太小就行了,比如我有500张图片,你这个纯数字要大于500)
以下代码可以直接套用
# -*- coding: utf-8 -*-
"""
Created on Mon May 18 17:42:17 2020
@author: 吴
"""
#批量修改文件名
#批量修改图片文件名
import os
import re
import sys
from PIL import Image
import string
PATH = r"C:/Users/吴/Desktop/学习快速通道/茶叶渣"
def renameall():
#待修改文件夹
fileList = os.listdir(PATH)
#输出文件夹中包含的文件
print("修改前:"+str(fileList))
#得到进程当前工作目录
currentpath = os.getcwd()
#将当前工作目录修改为待修改文件夹的位置
os.chdir(PATH)
num=1 #名称变量
for fileName in fileList: #遍历文件夹中所有文件
pat=".+\.(jpg|png|pgm|jpeg|JPG)" #匹配文件名正则表达式
pattern = re.findall(pat,fileName) #进行匹配
os.rename(fileName,(str(num)+'.'+pattern[0])) #文件重新命名
num = num+1 #改变编号,继续下一项
print("---------------------------------------------------")
os.chdir(currentpath) #改回程序运行前的工作目录
sys.stdin.flush() #刷新
print("修改后:"+str(os.listdir(PATH))) #输出修改后文件夹中包含的文件
dirName = r"C:/Users/吴/Desktop/学习快速通道/茶叶渣//" #最后要加双斜杠,不然会报错
li=os.listdir(dirName)
for filename in li:
newname = filename
newname = newname.split(".")
if newname[-1]=="jpeg":
newname[-1]="jpg"
newname = str.join(".",newname)
filename = dirName+filename
newname = dirName+newname
os.rename(filename,newname)
elif newname[-1] == 'png':
newname[-1]="jpg"
newname = str.join(".",newname)
filename = dirName+filename
newname = dirName+newname
os.rename(filename,newname)
print(newname,"updated successfully")
renameall()
此外,若想查看一个文件夹下多个子文件夹的图片处理,点这里~~