前言

在深度學習中，圖像的數據類型為torch，其形狀（shape）為：(C, H, W)。在opencv中圖像的數據類型為ndarray其形狀為：（H, W, C）
如果需要使用opencv顯示圖像則需要將torch轉化為ndarray; 如果需要使用該圖像作為深度學習的數據則需要將該圖像轉化為torch

np.ndarray轉為torch.Tensor

np.transpose( xxx,  (2, 0, 1))   # 將 H x W x C 轉化為 C x H x W

torch.Tensor轉numpy.ndarray

 np.transpose(tensor_cv.numpy(), (1, 2, 0)) # 將C x H x W 轉化為 H x W x C

參考

pytorch學習(五)—圖像的加載/讀取方式

免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。

猜您在找 torch.Tensor和numpy.ndarray 關於類型為numpy，TensorFlow.tensor，torch.tensor的shape變化以及相互轉化 Python中 list, numpy.array, torch.Tensor 格式相互轉化【python】list,tensor,numpy相互轉化 torch.Tensor和torch.tensor的區別 has invalid type , must be a string or Tensor torch.tensor(),torch.Tensor() TensorFlow中numpy與tensor數據相互轉化 [開發技巧]·TensorFlow中numpy與tensor數據相互轉化 torch.tensor的類型轉換以及和numpy的轉換