RenderedSST2

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

Rendered SST2 資料集.

Rendered SST2 是一個圖像分類資料集，用於評估模型的光學字元辨識能力。此資料集是透過轉譯 Standford Sentiment Treebank v2 資料集中的句子所產生。

此資料集包含兩個類別（正面和負面），並分為三個分割：一個包含 6920 張圖像的訓練分割（3610 張正面和 3310 張負面）、一個包含 872 張圖像的驗證分割（444 張正面和 428 張負面），以及一個包含 1821 張圖像的測試分割（909 張正面和 912 張負面）。

參數:

• root (str 或 pathlib.Path) – 資料集的根目錄。

• split (string, optional) – 資料集分割，支援 "train" (預設值), “val” 和 "test"。

• transform (callable, optional) – 一個函數/轉換，它接收一個 PIL 圖像並返回一個轉換後的版本。例如, transforms.RandomCrop。

• target_transform (callable, optional) – 一個函數/轉換，它接收目標並轉換它。

• download (bool, optional) – 如果為 True，則從網路下載資料集並將其放入根目錄。 如果資料集已經下載，則不會再次下載。 預設值為 False。

特殊成員:

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

參數:

index (int) – 索引

回傳:

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

回傳類型:

(Any)