ImageFolder

class torchvision.datasets.ImageFolder(root: ~typing.Union[str, ~pathlib.Path], transform: ~typing.Optional[~typing.Callable] = None, target_transform: ~typing.Optional[~typing.Callable] = None, loader: ~typing.Callable[[str], ~typing.Any] = <function default_loader>, is_valid_file: ~typing.Optional[~typing.Callable[[str], bool]] = None, allow_empty: bool = False)

一個通用資料載入器，預設情況下按以下方式組織影像

root/dog/xxx.png
root/dog/xxy.png
root/dog/[...]/xxz.png

root/cat/123.png
root/cat/nsdf3.png
root/cat/[...]/asd932_.png

此類繼承自 DatasetFolder，因此可以透過重寫相同的方法來自定義資料集。

引數:

• root (str 或 pathlib.Path) – 根目錄路徑。

• transform (可呼叫物件, 可選) – 一個函式/轉換，它接收 PIL 影像或 torch.Tensor（取決於給定的載入器），並返回轉換後的版本。例如，transforms.RandomCrop

• target_transform (可呼叫物件, 可選) – 一個函式/轉換，它接收目標並對其進行轉換。

• loader (可呼叫物件, 可選) – 一個根據影像路徑載入影像的函式。

• is_valid_file (可呼叫物件, 可選) – 一個函式，它接收影像檔案的路徑並檢查檔案是否有效（用於檢查損壞的檔案）

• allow_empty – 如果為 True，則空資料夾被視為有效類別。如果為 False（預設值），則空資料夾會引發錯誤。