Работа с MLMultiArray от MLModel

Я использую mlmodel в моем проекте для обнаружения объектов. Вход для модели Image (Color 300 x 300) и выходы MultiArray (Double) а также MultiArray (Double 4 x 1917 x 1), я сделал VNCoreMLRequest и я получил вывод, как указано выше. Здесь мне нужно показать имя объекта и доверие к пользователю из MLMultiArray выход. Теперь вопрос в том, как его расшифровать, как получить имя объекта и доверие от MLMultiArray, Я исследую последние 3 дня и не пользуюсь.

1 ответ

Решение

Число 1917 говорит мне, что вы, скорее всего, используете MobileNet + SSD.

Я написал несколько постов в блоге, которые объясняют, как это работает. Вот запись в блоге, которая объясняет, как выполнить декодирование для MobileNetV2+SSDLite, но та же логика работает и для других версий MobileNet и SSD: https://machinethink.net/blog/mobilenet-ssdlite-coreml/

Другие вопросы по тегам