我正在使用自定義數(shù)據(jù)集在 AWS sagemaker 上訓(xùn)練多類(lèi)圖像分類(lèi)模型。該數(shù)據(jù)集有大約 50 個(gè)類(lèi)。我正在關(guān)注這個(gè)筆記本：圖像分類(lèi)遷移學(xué)習(xí)演示根據(jù)我的理解，模型的最后一層輸出對(duì)應(yīng)于我們數(shù)據(jù)集中每個(gè)類(lèi)的概率。Sagemaker 希望以 mxnet recordio 的 .rec 格式提供數(shù)據(jù)集。由于我沒(méi)有手動(dòng)將標(biāo)簽轉(zhuǎn)換為單熱編碼，因此我不知道哪個(gè)層在輸出哪個(gè)類(lèi)的概率。我怎樣才能得到一個(gè)有序的類(lèi)列表，其中索引對(duì)應(yīng)于模型最后一層的輸出。甚至 AWS 提供的筆記本（上面的鏈接）也有硬編碼的有序列表（列表：object_categories）。轉(zhuǎn)換為 .rec 格式之前的數(shù)據(jù)集如下所示：./train/object1/   -image1.jpg   -image2.jpg   -image3.jpg   -...image500.jpg./train/object2/   -image1.jpg   -image2.jpg   -image3.jpg   -...image500.jpg..../train/object50/   -image1.jpg   -image2.jpg   -image3.jpg   -...image500.jpg任何幫助將不勝感激。