时间戳

首页 > 解决方案 > 如何从python中的不同文件夹和子文件夹加载数据（图像）

python - 如何从python中的不同文件夹和子文件夹加载数据（图像）

问题描述

我正在尝试数据集 DRISHTI-GS 上的问题。我想使用图像（来自训练）和杯段和磁盘段图像（来自软图）并在其上实现 CNN 模型。Ground Truth (GT) 子文件夹由 52 个子文件夹组成，每个子文件夹对应于另一个名为 Images 的文件夹（它是训练的子文件夹）中的 52 张图像。GT 中的子文件夹中有 3 个子文件夹，其中子文件夹 softmap 有两个图像，我需要加载这些图像。如果需要更多详细信息，我在下图中给出了子文件夹的布局。我希望有人可以帮助我提供一些具体的代码或一些有用的材料。

（基本上想在主目录及其所有子目录中列出和过滤所有扩展名为 .jpg 或 .png 的文件。）

子文件夹的布局

标签： pythonimagedirectory

解决方案

您可以使用stackabuse中的以下代码在当前目录中找到所有文件。

import os

for root, dirs, files in os.walk("."):
    for filename in files:
        print(filename)

如果你只想找到像文件这样的图像jpg文件，你可以像下面这样过滤它们：

import os

for root, dirs, files in os.walk("."):
    for filename in files:
       if filename.endswith(".jpg"):
           # Process the file

推荐阅读