SageMaker 图像分类：如何获取与模型输出对应的类的有序列表

Question

我正在使用自定义数据集在 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

任何帮助将不胜感激。

Answer 1

您需要使用 im2rec 从 lst 文件创建 recordio 文件。lst 文件是基于输入数据集创建的，标签的单热编码在 lst 文件中完成。有关如何创建多标签输入的示例，请参阅此笔记本。