1. 程式人生 > >python 檔案路徑(pytorch 資料集)

python 檔案路徑(pytorch 資料集)

最近學深度學習,發現需要跑資料集,資料通常存在data資料夾下。本文主要幫助那些看完程式碼後,不能正確安放資料集路徑的同學。

案例1: 

train_dataset = torchvision.datasets.MNIST(root='../../data', 
                                           train=True, 
                                           transform=transforms.ToTensor(),
                                           download=True)

root='../../data'不知道應該把資料集放在那裡????

案例2:

train_dataset = dsets.MNIST(root='./data', 
                            train=True, 
                            transform=transforms.ToTensor(),
                            download=True)

 root='./data',不知道把資料集放在那裡????

解決方案:直接檢視檔案路徑:

path1=os.path.abspath('.')
print(path1)
path2=os.path.abspath('../..')
print(path2)

執行結果:

E:\深度學習\深度學習框架Pytorch快速開發與實踐\chapter5
E:\深度學習