修改一个文件夹下含多个文件夹中的所有图片名称（代码）

喜欢ヅ旅行 2022-11-20 08:22 116阅读 0赞

#!/usr/bin/python
    # -*- coding:utf-8 -*-
     
    import os
    # 这步在创建文件夹xim,这个文件夹下面还有images文件夹和itst文件夹
    def mkdir(pathx):
     
    	p = os.path.exists(pathx)
     
    	if not p:                   #判断是否存在文件夹如果不存在则创建为文件夹
    		os.makedirs(pathx)            #makedirs 创建文件时如果路径不存在会创建这个路径
    		print("---  new p...  ---")
    		print("---  OK  ---")
     
    	else:
    		print("---  There is this p!  ---")
    		
    x = "/Users/sunchanglan/Desktop/ml/test/xim"
            #调用函数
    
    def generate(dir,label):
    	files = os.listdir(dir)#返回path指定的文件夹包含的文件或文件夹的名字的列表。
    	files.sort()
    	print('****************')
    	print('input :',dir)
    	print('start...')
    	listText = open('allof.txt','a')
    	for file in files:
    		fileType = os.path.split(file)
    		if fileType[1] == '.txt':
    			continue
    		name = file + ' ' + str(int(label)) +'\n'
    		listText.write(name)
    	listText.close()
    	print('down!')
    	print('****************')
    
    
    outer_path = '/Users/sunchanglan/Desktop/ml/test/xim'   #这里是你的图片的目录
    
    
    if __name__ == '__main__':
    	mkdir(x)     
    	i = 0
    	folderlist = os.listdir(outer_path)          #列举文件夹
    	print(folderlist)
    	for folder in folderlist:
    		generate(os.path.join(outer_path,folder),i)
    		# print(folder)
    		i += 1