Amazon SageMaker Неподдерживаемое приложение типа контента / x-image - PullRequest
Новая
       4

Amazon SageMaker Неподдерживаемое приложение типа контента / x-image

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

У меня есть модель CNN на основе тензорного потока / кераса, развернутая в Sagemaker.

Теперь, чтобы вызвать вывод, я следовал этому учебнику

Ниже фрагмент кода 

def inferImage(endpoint_name):
    # Load the image bytes
    img = open('./shoe.jpg', 'rb').read()
    runtime = boto3.Session().client(service_name='sagemaker-runtime')

    # Call your model for predicting which object appears in this image.
    response = runtime.invoke_endpoint(
        EndpointName=endpoint_name,
        ContentType='application/x-image',
        Body=bytearray(img))
    response_body = response['Body']
    print(response_body.read())

Когда я запускаю этот код, я получаю сообщение об ошибке

Unsupported content-type application/x-image

Чего мне не хватает?Любое предложение о том, как это исправить?

1 Ответ

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

Вы использовали SageMaker Python SDK?Если да, вы можете обратиться к этому README https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/tensorflow/deploying_python.rst и предоставить свой собственный input_fn () для работы с данными application / x-image.

Если вы не предоставите свой собственный input_fn () впользовательский скрипт, по умолчанию input_fn может обрабатывать только 3 типа: «application / json», «text / csv» и «application / octet-stream»

Здесь выдается исключение: https://github.com/aws/sagemaker-tensorflow-container/blob/1e74bc6440cdd7e083d15026869e021c5ab504a4/src/tf_container/serve.py#L239

...