python 檔案路徑(pytorch 資料集)
阿新 • • 發佈:2018-12-08
最近學深度學習,發現需要跑資料集,資料通常存在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:\深度學習