torchvision將圖片轉換成Tensor:transforms.ToTensor()

transforms.ToTensor() (1) transforms.ToTensor() 將numpy的ndarray或PIL.Image讀的圖片轉換成形狀爲(C,H, W)的Tensor格式,且/255歸一化到[0,1.0]之間 (2)通道的具體順序與cv2讀的還是PIL.Image讀的圖片有關係 cv2:(B,G,R) PIL.Image:(R, G, B) import torch i
相關文章
相關標籤/搜索