如何把pandas的dataframe資料載入到PyTorch的Dataloader中進行深度學習訓練?

參考資料

假設一個變數為df的dataframe有著以下的格式(如果class為文字格式,可透過下文所提供的程式碼進行encoding)

class | file_path
0 |000.jpg
1 | 111.jpg
2 | 222.jpg
3 | 333.jpg
4 | 444.jpg
5 | 555.jpg

如果想透過PyTorch進行影像分類模型訓練的話,我們可以透過以下的程式碼載入到dataloader中

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store