pytorch學習(二)—自定義數據集

#在深度學習中常常須要生成帶標籤的圖片名稱列表,xxxlist.txt文件, #編寫腳本語言,實現對文件中圖片生成帶標籤的txt文件方法 import os def generate(dir,label): files = os.listdir(dir) files.sort() print("*****************") print("input =
相關文章
相關標籤/搜索