python根目录的生成

python根目录的生成

import os
from os import getcwd

wd = getcwd()          #os.getcwd() 方法能够返回当前工作目录

datasets_path = "data/"        #里面有个文件夹data，图片就存于次文件夹中
photos_names = os.listdir(datasets_path)  #返回指定的文件夹包含的文件或文件夹的名字的列表
photos_names = sorted(photos_names)   #sorted() 函数对所有可迭代的对象进行排序操作

list_file = open('train_lines.txt', 'w')   #建立txt，把文件写进去
for photo_name in photos_names: 
    if(photo_name.lower().endswith(('.bmp', '.dib', '.png', '.jpg', '.jpeg', '.pbm', '.pgm', '.ppm', '.tif', '.tiff'))):# lower() 方法转换字符串中所有大写字符为小写
        #endswith(）判断一个字符串是否以特定的字符串后缀结尾，如果是则返回逻辑值True，否则返回逻辑值False。因为是if，所以true执行下面的操作
        list_file.write('%s/%s'%(wd, os.path.join(datasets_path, photo_name)))    #os.path.join()函数：连接两个或更多的路径名组件，中间会自动加“/”，末尾不加
        list_file.write('\n')                             #wb 只写打开或新建一个二进制文件；只允许写数据，跟w差不多，差别不大，“/n”表示空一行
list_file.close()             #写完，关闭