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** (字串, 可選) – 資料集拆分，支援 "train" (預設) 或 "test"。

• **transform** (可呼叫物件, 可選) – 一個接收 PIL 影像並返回變換後版本的函式/變換。例如，transforms.RandomCrop

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

特殊成員：

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

引數：

**index** (int) – 索引

返回：

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

返回型別：

(Any)