记录日常的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是子文件。