Python PyTorch GDriveReader用法及代碼示例

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

用法:

class torchdata.datapipes.iter.GDriveReader(source_datapipe: IterDataPipe[str], *, timeout: Optional[float] = None)

參數：

• source_datapipe-包含 GDrive 文件 URL 的 DataPipe

• timeout-HTTP 請求超時(以秒為單位)

獲取指向 GDrive 文件的 URL，並生成文件名和 IO 流的元組。

示例

>>> from torchdata.datapipes.iter import IterableWrapper, GDriveReader
>>> gdrive_file_url = "https://drive.google.com/uc?export=download&id=SomeIDToAGDriveFile"
>>> gdrive_reader_dp = GDriveReader(IterableWrapper([gdrive_file_url]))
>>> reader_dp = gdrive_reader_dp.readlines()
>>> it = iter(reader_dp)
>>> path, line = next(it)
>>> path
https://drive.google.com/uc?export=download&id=SomeIDToAGDriveFile
>>> line
<First line from the GDrive File>