【新手指南】Python中的listdir()函数的介绍

【新手指南】Python中的listdir()函数的介绍

在用pytorch导入dataset的时候对listdir()函数产生的是文件还是文件夹一直都有疑问，所以自己先在网上找了一段小代码调试，先小小的记录一下

一、打开的文件图

要打开的文件夹是蓝色框条选中的train训练集，该文件夹里有一些PNG格式的文件以及一个名为testcode的文件夹，文件夹里还包含一张名为test.png的图片，该文件的路径"E:\github\Semantic-segmentatio-master\CCF_remotesensing_dataset\train"

二、运行代码

代码里的路径要用绝对路径，也就是要用双//线，要导入os和sys模块：

>>>import os,sys
>>>path="E:\\github\\Semantic-segmentation-master\\CCF_remotesensing_dataset\\train\\"
>>>dirs=os.listdir(path)
>>>for file in dirs: 
>>>    print(file)

结果：

使用listdir()函数既可以得到该文件夹目录下的文件也可以得到文件夹，但是子文件夹里的内容分就访问不到了。