Python PyTorch RarArchiveLoader用法及代碼示例

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

用法:

class torchdata.datapipes.iter.RarArchiveLoader(datapipe: IterDataPipe[Tuple[str, io.BufferedIOBase]], *, length: int = - 1)

參數：

• datapipe-可迭代DataPipe提供路徑名和rar二進製流的元組

• length-DataPipe的標稱長度

從輸入可迭代數據管道中解壓縮 rar 二進製流，其中包含路徑名元組和 rar 二進製流，並生成路徑名元組和提取的二進製流(函數名稱： load_from_rar )。

注意

由於壓縮文件類型的限製，此 DataPipe 不支持嵌套 RAR 壓縮文件。請在閱讀內部壓縮文件之前解壓外部 RAR 壓縮文件。

示例

>>> from torchdata.datapipes.iter import FileLister, FileOpener
>>> datapipe1 = FileLister(".", "*.rar")
>>> datapipe2 = FileOpener(datapipe1, mode="b")
>>> rar_loader_dp = datapipe2.load_from_rar()
>>> for _, stream in rar_loader_dp:
>>>     print(stream.read())
b'0123456789abcdef'