记录日常的Python(一)—— os.walk遍历对文件夹中的所有图片进行访问
前言
此系列记录一下平常会用python处理的一些小问题
一、文件夹结构
我们先来看看,原来的文件夹的结构和有的东西。
我们要遍历的就是hymenoptera_data这个文件夹
此文件夹由如下组成
train和val里面都有ants和bees的文件夹
而ants和bees里面都是图片,我们要做的就是用os.walk对里面的图片进行查看。
一、利用os.walk对文件夹进行遍历
代码也是十分简单的,只需要调用一下os.walk即可
import os
import cv2 as cv
data_dir = "hymenoptera_data"
for root,dirs,files in os.walk(data_dir):
for name in files:
img_dir = os.path.join(root,name)
img = cv.imread(img_dir)
cv.imshow("image",img)
cv.waitKey()
#print(img_dir)
其中解释一下三个返回值root,dirs,files
参数 | 解释 |
---|---|
root | 所指的是当前正在遍历的这个文件夹的本身的地址 |
dirs | 文件夹中所有的目录的名字(不包括子目录) |
files | 该文件夹中所有的文件(不包括子目录) |
比较通俗的说就是,root是根目录,dirs是子目录,files是子文件。