Python PyTorch ImageFolder用法及代碼示例

本文簡要介紹python語言中 torchvision.datasets.ImageFolder 的用法。

用法:

class torchvision.datasets.ImageFolder(root: str, transform: Optional[Callable] = None, target_transform: Optional[Callable] = None, loader: Callable[[str], Any] = <function default_loader>, is_valid_file: Optional[Callable[[str], bool]] = None)

參數：

• root(string) -根目錄路徑。

• transform(可調用的,可選的) -接受 PIL 圖像並返回轉換版本的函數/轉換。例如，transforms.RandomCrop

• target_transform(可調用的,可選的) -接收目標並對其進行轉換的函數/轉換。

• loader(可調用的,可選的) -在給定路徑的情況下加載圖像的函數。

• is_valid_file-獲取圖像文件路徑並檢查文件是否為有效文件的函數(用於檢查損壞的文件)

默認情況下以這種方式排列圖像的通用數據加載器：

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，因此可以重寫相同的方法來自定義數據集。