我正在使用自定義數據集在 AWS sagemaker 上訓練多類圖像分類模型。該數據集有大約 50 個類。我正在關注這個筆記本：圖像分類遷移學習演示根據我的理解，模型的最后一層輸出對應于我們數據集中每個類的概率。Sagemaker 希望以 mxnet recordio 的 .rec 格式提供數據集。由于我沒有手動將標簽轉換為單熱編碼，因此我不知道哪個層在輸出哪個類的概率。我怎樣才能得到一個有序的類列表，其中索引對應于模型最后一層的輸出。甚至 AWS 提供的筆記本（上面的鏈接）也有硬編碼的有序列表（列表：object_categories）。轉換為 .rec 格式之前的數據集如下所示：./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任何幫助將不勝感激。