1. 程式人生 > >pytorch中的包匯入,真心記不住啊!!!

pytorch中的包匯入,真心記不住啊!!!

datasets.ImageFolder(os.path.join(data_dir,'train'+train_all),data_transforms['train']);

上面的最後一行,這裡的資料打包的處理方式很有意思。是放進一個字典中去,到時候呼叫就好。真的很方便。程式碼直觀。

有路徑有轉換的的語句,這裡轉換的關鍵字transform省略了。另外,compose的命令是打包這個轉換的:

打包的內容是個列表,也就是說compose([  ])是這樣的處理方式。另外記住的是,compose是來自於:

,所以這裡記住一下,

ImageFolder,compose,DataLoader來自於哪個包裡。一般的求導啊,autograd  優化 optim  資料打包 DataLoader都是來自torch的。