SBDataset

class torchvision.datasets.SBDataset(root: Union[str, Path], image_set: str = 'train', mode: str = 'boundaries', download: bool = False, transforms: Optional[Callable] = None)

語義邊界資料集

SBD 當前包含來自 PASCAL VOC 2011 資料集的 11355 張影像的標註。

注意

請注意，此資料集包含的 train 和 val 劃分與 PASCAL VOC 資料集中的劃分不同。特別是，一些“train”影像可能是 VOC2012 val 的一部分。如果您有興趣在 VOC 2012 val 上進行測試，請使用 image_set=’train_noval’，這會排除所有 val 影像。

警告

此類需要 scipy 來載入 .mat 格式的目標檔案。

引數：

• root (str 或 pathlib.Path) – 語義邊界資料集的根目錄

• image_set (string, 可選) – 選擇要使用的 image_set，train、val 或 train_noval。Image set train_noval 排除 VOC 2012 val 影像。

• mode (string, 可選) – 選擇目標型別。可能的值為 ‘boundaries’ 或 ‘segmentation’。如果是 ‘boundaries’，目標是一個形狀為 [num_classes, H, W] 的陣列，其中 num_classes=20。

• download (bool, 可選) – 如果為 True，則從網際網路下載資料集並將其放在根目錄中。如果資料集已下載，則不會再次下載。

• transforms (callable, 可選) – 一個函式/變換，它將輸入樣本及其目標作為輸入，並返回變換後的版本。輸入樣本是 PIL 影像；如果 mode=’boundaries’，目標是 numpy 陣列；如果 mode=’segmentation’，目標是 PIL 影像。

特殊成員：

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

引數：

index (int) – 索引

返回：

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

返回型別：

(Any)