#!/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