Вопрос: загрузка модели с помощью пользовательского видения и API Python

Я новичок в Microsoft Custom Vision, и я работаю над интеграцией Microsoft Azure Custom Vision API с использованием jupyter notebooks / python. Я смог загрузить изображения, автоматически пометить их и обучить первые итерации. Однако, когда я пытался загрузить файл Docker итерации / модели поезда, я застрял при попытке экспортировать модель. Используя функцию export_iteration, я получил объект mst.rest.pipeline.clientrawresponse. Я думаю, что в настоящее время он хранится только в очереди на экспорт. Как мне получить доступ к этому элементу очереди, чтобы загрузить его в мою локальную систему?

PS: Я работаю с общим (компактным) форматом модели, поэтому его можно экспортировать.

Пример кода:

# Initalize the Training Client 
training_key = "your-training-key"
ENDPOINT = "your-endpoint" 

c_plat = CustomVisionTrainingClient(training_key,ENDPOINT)
# List all projects you have
projects = c_plat.get_projects()

#Always take the newest project and its newest iteration and export it 
iterations = c_plat.get_iterations(projects[0].id)
c_plat.export_iteration(project_id=projects[0].id, iteration_id=iterations[0].id, platform = "DockerFile", raw=True, flavor = "ARM")


person ruki ami    schedule 08.04.2020    source источник


Ответы (1)


После проб и ошибок я нашел решение:

    #Always takes the newest project and its newest iteration
    iterations = c_plat.get_iterations(projects[0].id)
    response = c_plat.export_iteration(project_id=projects[0].id, iteration_id=iterations[0].id, platform = "DockerFile", raw=False, flavor="ARM")

    # Opnening the uri
    import webbrowser
    webbrowser.open(c_plat.get_exports(project_id=projects[0].id, iteration_id=iterations[0].id)[0].download_uri)

Это открыло URI в новой вкладке и запустило автоматическую загрузку. Надеюсь, это поможет кому-то другому.

Ваше здоровье!

person ruki ami    schedule 08.04.2020