FER2013

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

FER2013 資料集。

注意

只有當 fer2013.csv 或 icml_face_data.csv 存在於 root/fer2013/ 中時，此資料集才能傳回測試標籤。 如果只有 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 (callable, optional) – 一個函數/轉換，它接收目標並轉換它。

特殊成員:

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

參數:

index (int) – 索引

傳回:

樣本和元數據，可選擇透過各自的轉換進行轉換。

傳回類型:

(Any)