利用PyTorch自定義數據集實現貓狗分類

看了許多關於PyTorch的入門文章，大抵是從torchvision.datasets中自帶的數據集進行訓練，致使很難把PyTorch運用於本身的數據集上，真正地靈活運用PyTorch。python 這裏我採用從Kaggle上下載的貓狗數據集，利用自定義數據集訓練本身的二分類神經網絡。網絡 解壓後，一個文件裏面有12500張圖，貓狗各一半，文件名相似於這樣：cat.0.jpg、dog.12499.

>>阅读原文<<