Python修改文件夹中图片名并对其进行移动

最近搞泰迪杯,数据增强的时候文件生成目录出了些问题,又不想重新增强,所以写个demo解决一下我的问题。

先看一下我的文件格式,这样的文件夹有40个,每个下边都储存了增强后的image和mask,而且名字乱乱的。
Python修改文件夹中图片名并对其进行移动_第1张图片
Python修改文件夹中图片名并对其进行移动_第2张图片
例如文件夹/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)

最后结果
Python修改文件夹中图片名并对其进行移动_第3张图片

Python修改文件夹中图片名并对其进行移动_第4张图片
这样标签和原图就都是一一对应的储存在两个文件夹下了。

你可能感兴趣的:(数据处理)