1. 程式人生 > >用pytorch實現一個神經網路(一)

用pytorch實現一個神經網路(一)

對於影象資料的resize問題:
pytorch裡有幾種resize資料的方法:
1.torchvision.transforms.Resize:這個我始終沒用成,好像是伺服器上安裝的anaconda包裡的函式不太對。這個函式是對於PIL影象進行操作的。
2.對於numpy.ndarray的資料resize,函式為o=np.resize(i,newSize)
這個函式我試了,輸出的結果再儲存成影象就不對了。
3.PILimage自帶的resize函式。o=i.resize(newSize,interpolation)
這個也是對PIL影象進行操作的。

那我的需求實際上是要對生成的tensor資料進行操作,但是始終沒找到啊。對於numpy資料操作的那個,似乎三通道的就不對了