Перейти к основному содержимому

Создание визуального инференсера CoreML (coreml.new_vision_request)

Объявление

объект_визуального_инференсера, сообщение_об_ошибке = coreml.new_vision_request(путь_к_скомпилированной_модели)

или

объект_визуального_инференсера, сообщение_об_ошибке = coreml.new_vision_request({
compiled_model_path = путь_к_скомпилированной_модели,
uses_cpu_only = использовать_только_CPU,
compute_units = конфигурация_вычислительных_блоков,
image_crop_and_scale_option = режим_кадрирования_и_масштабирования,
})

Параметры

  • путь_к_скомпилированной_модели
    Текстовое значение, путь к скомпилированной модели; сейчас поддерживаются модели классификации изображений и обнаружения объектов

  • uses_cpu_only
    Логическое значение, необязательный параметр, использовать ли по умолчанию только CPU для инференса, по умолчанию false

  • compute_units Текстовое значение, действует в iOS 12+. Поддерживаются следующие значения:

    • "all"
    • "cpu_only" или "cpu"
    • "cpu_and_gpu" или "gpu"
    • "cpu_and_neural_engine", "ane" или "neural_engine" (iOS 16+)
  • image_crop_and_scale_option
    Целое число, необязательный параметр, режим кадрирования и масштабирования при визуальном инференсе

Возвращаемое значение

  • объект_визуального_инференсера
    Объект визуального инференсера; при ошибке создания возвращается nil

  • сообщение_об_ошибке
    Текстовое значение; при успехе возвращается nil, при ошибке — сообщение об ошибке

Описание

  • Доступно в версиях ПО 1.3.8 и выше
  • Не поддерживается в системах ниже iOS 13
  • Загружает скомпилированную модель .mlmodelc и возвращает контекст для инференса изображений
  • Предназначено для визуальных моделей CoreML, принимающих изображение на вход
  • Если вход модели представляет собой универсальный признак image / MLMultiArray, можно рассмотреть coreml.new_model_request
  • Помимо :predict() / :run(), объект поддерживает методы запроса :results(), :is_done(), :compute_units(), :image_crop_and_scale_option(), :class_labels() и другие

Пример

См. инференс изображения с помощью визуального инференсера CoreML