快捷方式

encode_png

torchvision.io.encode_png(input: Tensor, compression_level: int = 6) Tensor[原始碼]

接收 CHW 佈局的輸入張量,並返回對應 PNG 檔案的內容的緩衝區。

引數:
  • input (Tensor[通道, 影像高度, 影像寬度]) – int8 影像張量,具有 c 個通道,其中 c 必須為 3 或 1。

  • compression_level (int) – 結果檔案的壓縮因子,它必須是 0 到 9 之間的數字。預設為:6

返回:

一個包含 PNG 檔案原始位元組的一維 int8 張量。

PNG 檔案。

返回型別:

Tensor[1]

文件

訪問全面的 PyTorch 開發者文件

檢視文件

教程

為初學者和高階開發者提供深入的教程

檢視教程

資源

查詢開發資源並讓您的問題得到解答

檢視資源