Python PyTorch FileLister用法及代碼示例

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

用法:

class torchdata.datapipes.iter.FileLister(root: Union[str, Sequence[str], IterDataPipe] = '.', masks: Union[str, List[str]] = '', *, recursive: bool = False, abspath: bool = False, non_deterministic: bool = False, length: int = - 1)

參數：

• root-根目錄或一係列根目錄

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

• recursive-是否從嵌套目錄返回路徑名

• abspath-是返回相對路徑名還是絕對路徑名

• non_deterministic-是否按排序順序返回路徑名。如果 False ，每個根目錄產生的結果將被排序

• length-數據管道的標稱長度

給定根目錄的路徑，生成根目錄中文件的文件路徑名(路徑 + 文件名)。可以提供多個根目錄。

示例

>>> from torchdata.datapipes.iter import FileLister
>>> dp = FileLister(root=".", recursive=True)
>>> list(dp)
['example.py', './data/data.tar']