浏览和加载来自不同子文件夹 Python 的数据

如何解决浏览和加载来自不同子文件夹 Python 的数据

对于图像处理任务，我想从给定目录加载数据。 该目录由 3 个主要类别（文件夹）组成，每个类别有几个子类（文件夹），每个子类也可以有多个子类（文件夹）。 首先，我想按主要 3 个类别加载所有图像。因此，我需要该算法从不同的子文件夹中提取图像并将其添加到训练数据中。

我使用了以下代码：

datadir = r'C:\xxx'        
categories = ['a','b','c']

    training_data = []

    
    def create_training_data():
       for category in categories:
            path = os.path.join(datadir,category)
            class_num = categories.index(category)
            for img in os.listdir(path):
                try:
                    img_array=cv2.imread(os.path.join(path,img))
                    training_data.append([img_array,class_num])
                except Exception as e:
                    pass

我尝试过使用 os.walk 来实现这个需求，但没有成功。

感谢您的帮助。