您的位置:首页 > 编程语言 > Python开发

利用python.os.listdir获取文件夹中所有文件名

2017-12-16 19:26 543 查看
利用内置os的listdir可以获取文件夹中的文件名,从而可以利用文件名对数据做进一步的处理,比如:在猫狗识别的练习过程中,需要通过文件名对猫狗数据进行分类。

def get_file(file_dir):
dogs = []
cats = []
dog_labels = []
cat_labels = []
#利用listdir可以获取文件夹中所有文件的名称
for dir in os.listdir(file_dir):
temp = dir.split(".")
if temp[0] == 'dog':
dogs.append(file_dir+dir)
dog_labels.append(1)
else:
cats.append(file_dir+dir)
cat_labels.append(0)
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  tensorflow listdir