如何從python中的不同文件夾和子文件夾加載數據（圖像）

Question

我正在嘗試數據集 DRISHTI-GS 上的問題。 我想使用圖像（來自訓練）和cup seg 和disc seg 圖像（來自softmap）並在其上實現CNN 模型。 Ground Truth(GT) 子文件夾由 52 個子文件夾組成，每個子文件夾對應於另一個名為 Images 的文件夾中的 52 張圖像（這是訓練的子文件夾）。 我已經給出了子文件夾的布局。 GT 中的子文件夾中有 3 個子文件夾，其中子文件夾 softmap 有兩個圖像，我需要加載這些圖像。 我希望有人可以幫助我提供一些具體的代碼或一些有用的材料。

（基本上想列出和過濾主目錄及其所有子目錄中的所有文件，擴展名為 .jpg 或 .png。 ）

Answer 1

您可以使用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

Answer 2

如果我正確理解了您的問題，我認為這就是您正在尋找的東西（尚未對此進行測試，但它應該可以工作）

import glob

paths = glob.glob("DRISHTI_GS/Drishti-GS1_files/Training/GT/**/SoftMap/*.png")
print(paths)

這將產生一個字符串列表，其中每個字符串都是圖像的路徑。 此相對路徑模式假定DRISHTI_GS位於當前工作目錄中。 你也可以給它一個絕對路徑。

glob-pattern 中的**表示所有直接子文件夾，無論它們的名稱如何。

*.png代表任何名稱的圖像。 在這種情況下，它只會捕獲 PNG 圖像。 您沒有指定圖像的格式 - 如果它們不是 PNG，只需將模式更改為*.jpg等。