SageMaker Image Classification: Как получить упорядоченный список классов, соответствующий выводу модели - PullRequest
Новая
       42

SageMaker Image Classification: Как получить упорядоченный список классов, соответствующий выводу модели

0 голосов
/ 28 декабря 2018

Я тренирую модель для многоклассовой классификации изображений на AWS sagemaker, используя пользовательский набор данных.Набор данных насчитывает около 50 классов.Я слежу за этой записной книжкой: Демонстрация обучения передаче классификации изображений

Согласно моему пониманию, последний слой модели выводит вероятности, соответствующие каждому классу в нашем наборе данных.Sagemaker ожидает, что набор данных будет предоставлен в формате .rec для mxnet recordio.Поскольку я не конвертирую метки вручную в одно горячее кодирование, я не знаю, какой уровень выводит вероятности для какого класса.Как получить упорядоченный список классов, индексы которых соответствуют выходным данным последнего слоя модели.

Даже в ноутбуке, предоставленном AWS (ссылка выше), этот упорядоченный список (list: 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

Любая помощь будет высоко оценена.

Ответы [ 2 ]

0 голосов
/ 23 февраля 2019

Вам необходимо использовать im2rec, чтобы создать файл записи из файла lst.Файл lst создается на основе входного набора данных, и в файле lst выполняется однократное кодирование меток.Пожалуйста, обратитесь к этому блокноту для примера того, как создать ввод с несколькими метками.

0 голосов
/ 31 декабря 2018

Метки встроены в файл recordio .rec, поэтому для пользовательских приложений с несколькими метками вам придется повторно маркировать.Откровенно говоря, документация не супер, но вот отправная точка (перейдите к основанию): https://mxnet.incubator.apache.org/faq/recordio.html

Что касается слоя с метками, то последний слой - это то, что генерирует вероятность метки.Архитектура модели абстрагируется в Sagemaker, а установка гиперпараметра класса гарантирует, что последний уровень назначит вероятности для числа классов, найденных в файле .rec.

...