捷徑

Utils

torchvision.utils 模組包含各種實用工具,主要是 用於視覺化

draw_bounding_boxes(image, boxes[, labels, ...])

在給定的 RGB 影像上繪製邊界框。

draw_segmentation_masks(image, masks[, ...])

在給定的 RGB 影像上繪製分割遮罩。

draw_keypoints(image, keypoints[, ...])

在給定的 RGB 影像上繪製關鍵點。

flow_to_image(flow)

將 flow 轉換為 RGB 影像。

make_grid(tensor[, nrow, padding, ...])

建立影像網格。

save_image(tensor, fp[, format])

將給定的 Tensor 儲存到影像檔案中。

文件

存取 PyTorch 的完整開發人員文件

檢視文件

教學課程

取得初學者和進階開發人員的深入教學課程

檢視教學課程

資源

尋找開發資源並獲得問題解答

檢視資源