我就廢話不多說了，大家還是直接看代碼吧！
#!/usr/bin/python # -*- coding: UTF-8 -*- from PIL import Image import numpy as np import torch pil_img = Image.open('/Users/songlu/Desktop/code.jpg') img = np.array(pil_img) print torch.from_numpy(img)
補充知識：pytorch mxnet 多GPU訓練 batch 的區別
區別有很多，這是一點點不一樣
1、mxnet 的batch ，設置的是單卡的，多卡是 batch*多卡，多一張卡，速度可以提升一倍
2、pytorch 是 batch/多卡數量，分配給每個gpu,設置的batch是總的batch
[ljg58026 ] Pytorch使用PIL和Numpy將單張圖片轉為Pytorch張量方式已經有235次圍觀