Развернуть обученную модель обнаружения объектов Sagemaker Mxnet локально? - PullRequest
0 голосов
/ 02 мая 2019

Я обучил модели обнаружения объектов в AWS Sagemaker.Я хочу использовать эту модель локально в моей машине.Я скачал эту модель, которая состоит из 3 файлов: hyperparams.json, model-symbol.json и model-0000.params.Я видел множество учебных пособий для локального развертывания модели классификации объектов, но не нашел ни одного для обнаружения объектов.

Ответы [ 2 ]

0 голосов
/ 04 мая 2019

Как упомянул NRauschmayr, может быть удобно использовать функцию локального режима Python SDK в SageMaker, если вы также используете ее для обучения в SageMaker. Одно небольшое исправление заключается в том, что вы должны установить для instance_type предиктора значение «local», поскольку вы заинтересованы в локальном развертывании. Подробнее о локальном режиме вы можете прочитать здесь: https://sagemaker.readthedocs.io/en/stable/overview.html#local-mode

0 голосов
/ 03 мая 2019

Существуют различные возможности:

  • Вы можете установить SageMaker Python SDK локально на своей машине и использовать SageMaker для обучения и развертывания своей модели локально. Поэтому вам нужно указать в MXNet Estimator train_instance_type = 'local».Затем SageMaker загрузит данные с S3 на вашу локальную машину и обучит модель там.Затем mxnet_estimator.deploy развернет модель локально.Вы можете найти подробное описание здесь
  • Вы можете использовать модель сервера MXNet.Вам необходимо преобразовать файлы модели в архив модели (.mar). Развернуть ее можно следующим образом: mxnet-model-server --models mymodel=mymodel.mar
...