在做cv相关题目时,不可避免的会涉及到对图片进行上下左右翻转的操作,但如果碰到图片名称没有按照顺序命名的话,新生成的图片命名就是一个问题。以下是我的代码实现。
1.引入对应的包和库
from PIL import Image #图像处理库
import os # OS库import shutil # shutil库(可用于文件或者图片复制到指定文件夹)
2.设置待处理图片的文件夹,以及保存新生成的图片的文件夹
ori='C:/Users/sanfrancisco/Desktop/piliang/6912/' #原图片所在文件夹
last='C:/Users/sanfrancisco/Desktop/piliang/last/' #新生成图片所在的文件夹
3.遍历得到文件夹下面所有名称
list_img = os.listdir(dir_img) #得到文件夹下面所有文件的名称
4.批量打开文件并进行翻转操作再保存至响应的文件夹
j = 3813
for img_name in list_img: #遍历获取每一个图片的名称
j = j+1
pri_image = Image.open(ori + img_name)
img_name='0'+str(j)+'.jpg' # 新的文件名
tmppath = last+ img_name #新的保存路径 pri_image.transpose(Image.FLIP_TOP_BOTTOM).save(tmppath) #将图片上下翻转并保存到tmppth这个路径下面
5.标签修改,以上下翻转为例子
import xml.etree.ElementTree as ET #引入处理xml文件的一个类
for i in range(4390, 4966):
path1 = 'C:/Users/sanfrancisco/Desktop/xml/'+'0'+str(i)+".xml"
tree = ET.parse(path1) #读取路径1
root = tree.getroot() #获得根节点
objects = root.findall("object") # 找到xml文件中所有的object属性块
for obj in objects: #遍历每一个object块
bbox = obj.find('bndbox') #找到每一个object块中bndbox属性
y1 = int(bbox.find('ymin').text) # 获取bndbox块中ymin和ymax属性的值(数据类型转化为为整形)
y2 = int(bbox.find('ymax').text)
a = 3648 - y2 #以上下翻转为例,只有y相关的两个值会改变
b = 3648 - y1
y1 = a
y2 = b
bbox.find('ymin').text = str(int(y1)) #将改变以后的值重新赋值给ymin和ymax属性
bbox.find('ymax').text = str(int(y2))
tree.write(path1) #按照原路径保存修改好的xml文件