图片批量改名+图片数据集生成.txt文件的python代码实现

---

图片批量改名

import os
path_name='D:/数据集/ships_classification/val/s9_val' # 需要批量修改的文件夹目录
i=1
for item in os.listdir(path_name):  
    os.rename(os.path.join(path_name,item),os.path.join(path_name,(str(i)+'.jpg'))) # 
    i+=1

修改后的效果：

图片数据集生成.txt文件：

import os
def generate(dir, label):
    files = os.listdir(dir)
    files.sort()
    listText = open('1_val.txt', 'a')        # 已经生成好的数据
    for file in files:
        fileType = os.path.split(file)
        if fileType[1] == '.txt':
            continue
        name = ' /root/data/office/new_val/'+'s'+str(int(label))+'/' +file + ' ' + str(int(label)) + '\n'      # .txt内的形式
        listText.write(name)
    listText.close()    
outer_path = 'D:/数据集/new_ships_classification/new_val'  # 图片的目录
if __name__ == '__main__':
    i = 0
    folderlist = os.listdir(outer_path)  
    for folder in folderlist:
        generate(os.path.join(outer_path, folder), i)
        i += 1

生成效果：可以对照下方图片的内容修改相应代码