用python自動生成train，val文件——pytorch數據集

任務——分類 數據集爲5個不同類別的圖片集，每個圖片集大概有3W張圖片。所以要建立一個train訓練的txt文件和一個val驗證的txt文件，裏面放圖片的路徑，因爲只是練手用，所以不放test驗證。 最終要的結果是從每個文件裏拿出28000個訓練和剩下差不多3000個用來測試。 import os a=0 while(a<5): dir = '/home/zyx/data/pic/'+s

>>阅读原文<<