Python PyTorch IoPathFileLister用法及代碼示例

本文簡要介紹python語言中 torchdata.datapipes.iter.IoPathFileLister 的用法。

用法:

class torchdata.datapipes.iter.IoPathFileLister(root: str, masks: Union[str, List[str]] = '', *, pathmgr=None)

參數：

• root-列出文件的根本地文件路徑或 URL 目錄

• masks-用於過濾文件名的 Unix 樣式過濾器字符串或字符串列表

• pathmgr-自定義 iopath.PathManager 。如果未指定，則創建默認 PathManager。

在提供的 root 路徑名或 URL 處列出目錄的內容，並為目錄中的每個文件生成完整的路徑名或 URL。

注意

默認PathManager當前支持本地文件路徑、普通HTTP URL和OneDrive URL。僅當 iopath >=0.1.9 時才支持 S3 URL。

示例

>>> from torchdata.datapipes.iter import IoPathFileLister
>>> datapipe = IoPathFileLister(root=S3URL)