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']