最近搞泰迪杯,数据增强的时候文件生成目录出了些问题,又不想重新增强,所以写个demo解决一下我的问题。
先看一下我的文件格式,这样的文件夹有40个,每个下边都储存了增强后的image和mask,而且名字乱乱的。
例如文件夹/home/csj/task/task1/output/8中都是如下的形式:
·········
groundtruth(1)_8_8.jpg_000772b0-5fd4-4d11-b1a1-4d7010805447.jpg
8_original_8.jpg_000772b0-5fd4-4d11-b1a1-4d7010805447.jpg
·········
我想将/home/csj/task/task1/output下原始image与掩模图mask分开重命名再储存到/home/csj/task/task1文件夹中的train和label中。
上代码,先将mask的后缀改为png,然后分割字符串匹配改名字移动到对应目录下。
import os
path = r"/home/csj/task/task1/output"
for pa in os.listdir(path):
pathdir = "/home/csj/task/task1/output/"+pa
for file in os.listdir(pathdir):
ext = os.path.splitext(file)
#print(str(ext).split('_')[-1])
#print(str(ext).split('_')[1]) #groundtruth or original
#print(str(ext[0]).split('_')[-1])
if str(ext).split('_')[1] == 'groundtruth':
newext = '.png'
filename = 'label' + str(ext[0]).split('_')[-1] + '.png'
oldfile = os.path.join(pathdir,file)
newfile = os.path.join(r"/home/csj/task/task1/label",filename)
os.rename(oldfile,newfile)
for file in os.listdir(pathdir):
ext = os.path.splitext(file)
if ext[1] == '.jpg':
filename = 'train' + str(ext[0]).split('_')[-1] + '.jpg'
oldfile = os.path.join(pathdir,file)
newfile = os.path.join(r"/home/csj/task/task1/train",filename)
os.rename(oldfile,newfile)
最后结果
这样标签和原图就都是一一对应的储存在两个文件夹下了。