В модели экспорта когнитивной службы Azure CustomVision в TensorFlow отсутствуют файлы - PullRequest
0 голосов
/ 11 июня 2019

Я использую Azure Cognitive Services, также известный как веб-сайт CustomVision, для создания, обучения и тестирования моделей. Я понимаю, что основная цель этого сайта - создать API, который можно вызвать для запуска вашей модели в производство. Я должен упомянуть, что я использую это, чтобы сделать обнаружение объекта.

Бывают случаи, когда вам нужно поддерживать автономный запуск (т. Е. У вас нет подключения к Azure и т. Д.). Я полагаю, что Microsoft знает и понимает это, потому что у них есть функция, которая позволяет вам экспортировать вашу модель во множество различных форматов (таких как TensorFlow, ONNX и т. Д.).

Проблема, с которой я сталкиваюсь, особенно связана с тем, что вы экспортируете в TensorFlow, который мне нужен, он загружает только замороженный график модели (model.pb). Однако бывают случаи, когда вам нужен либо файл .pbtxt, который идет вместе с моделью, либо файл конфигурации. Я знаю, что вы можете сгенерировать файл pbtxt, но для этого вам нужен .config.

Кроме того, после экспорта вы почти не получаете никакой информации о вашей модели, например, какой должен быть размер входного изображения. Я хотел бы видеть это лучше задокументировано где-нибудь. Например, 300x300 и т. Д. Не имея конфигурации или pbtxt вместе с моделью, вы должны выяснить это, загрузив вашу модель в TensorBoard или что-то подобное, чтобы выяснить входную информацию (размер, имя и т. Д.). ..). Более того, мы даже не знаем, какова базовая модель, ResNet, SSD и т. Д.

Итак, кто-нибудь знает, как я могу получить эти недостающие файлы при экспорте модели? Или кто-нибудь знает, как вы можете сгенерировать pbtxt, когда у вас есть только замороженный граф .pb файл?

Если нет, я бы порекомендовал эти улучшения для группы служб Azure Cognitive. Со всеми этими недостающими данными или информацией очень трудно использовать экспортированную модель.

Спасибо!

Ответы [ 2 ]

0 голосов
/ 13 июня 2019

Многие модельные архитектуры позволяют вам изменять размер сетевого ввода, например Yolo, который является архитектурой, экспортированной из Custom Vision.Включение фиксированного размера ввода не имеет смысла в этом случае.

Netron будет вашим хорошим другом и довольно простым в использовании, чтобы выяснить подробности о модели.

0 голосов
/ 11 июня 2019

Служба Custom Vision Service экспортирует только компактные домены. Для экспорта обнаружения объектов существует код для загрузки и запуска модели обнаружения объектов в загруженном zip-файле (model.pb, tags.txt).Наряду с экспортной моделью вы найдете код Python для тренировки модели.

...