os.path模塊中常用方法os.path.dirname與os.path.basename的結合

最近在做卷積神經網絡,對於數據集,我們經常獲取到的是下圖中的格式。文件夾名是標籤(類名),文件夾內是對應標籤的圖片 我們最好對數據集進行整理,把每張圖與標籤存成txt文件或者tfrecord,方便後面網絡訓練。 如果我們知道一張圖片的絕對路徑path,那麼我們就可以用 label = os.path.basename(os.path.dirname(path)) 獲得這個圖片的label。 os.
相關文章
相關標籤/搜索