FER2013

class torchvision.datasets.FER2013(root: Union[str, Path], split: str = 'train', transform: Optional[Callable] = None, target_transform: Optional[Callable] = None)

FER2013 資料集。

注意

如果 root/fer2013/ 中存在 fer2013.csv 或 icml_face_data.csv，則此資料集只能返回測試標籤。如果只存在 train.csv 和 test.csv，則測試標籤將設定為 None。

引數:

• root (str 或 pathlib.Path) – 資料集根目錄，其中存在目錄 root/fer2013。此目錄可能包含 fer2013.csv、icml_face_data.csv，或 train.csv 和 test.csv。優先順序按此順序，即如果存在 fer2013.csv，則其餘檔案將被忽略。所有這些（組合）檔案包含相同的資料，併為了方便起見而支援，但只有 fer2013.csv 和 icml_face_data.csv 能夠返回非 None 的測試標籤。

• split (string, optional) – 資料集分割，支援 "train" (預設) 或 "test"。

• transform (callable, optional) – 一個函式/變換，接受 PIL 影像並返回變換後的版本。例如，transforms.RandomCrop

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

特殊成員:

__getitem__(idx: int) → tuple[Any, Any]

引數:

index (int) – 索引

返回:

樣本和元資料，可透過相應的變換進行可選變換。

返回型別:

(Any)