Как преобразовать модель стилизации производителя TensorFlow в Core ML для Swift? ⇐ IOS

[Anonymous]

Я пытаюсь реализовать произвольную стилизацию изображения в Swift, используя предварительную модель Tensorflow. После загрузки модели я получил файл save_model.pb. Тем не менее, iOS не поддерживает этот формат напрямую, поэтому я попытался преобразовать его в Core ML с использованием Coremltools.
код Python для преобразования:

Код: Выделить всё

import coremltools as ct

# Load the TensorFlow model
model = ct.convert(
"/Users/kirit/Downloads/saved_model",  # e.g., a SavedModel directory
inputs=[
ct.ImageType(shape=(1, 256, 256, 3), name="contentImage", scale=1/255),
ct.ImageType(shape=(1, 256, 256, 3), name="styleImage", scale=1/255)
],
outputs=[ct.TensorType(name="stylizedImage")],
source="tensorflow"
)

# Save the Core ML model
model.save("DualInputStylization.mlmodel")

Проблема:

• Процесс преобразования всегда терпит неудачу с различными ошибками.
  Общие ошибки включают: < /li>
  Не удалось загрузить _mlmodelproxy: нет модуля с именем coremltools.libcoremlpython '< /li>
  Конвертер не был вызван с source = "tensorflow", но отсутствует пакет Tensorflow < /li>
  ошибки импорта TensorFlow из -за отсутствующего модуля IMP в Python 3.13 < /li>
  < /ul>
  система системы: < /strong>
  версия Python: 3.13.1
  Tensorflow версия: (последняя)
  версия Coremltools: (Последние)
  OS: macOS ( 15.1.1 (24b91)) < /p>
  Вопросы: < /strong> < /p>
  
  Как я могу успешно преобразовать TensorFlow Saved_model.pb для произвольной стилизации изображения в основном формате ML MLModel? < /li>
  Есть ли альтернативный подход для использования этой модели в Swift? Есть ли определенная версия Tensorflow, которая лучше всего подходит для Core ML преобразования?
  Любое руководство будет оценено! Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/794 ... -for-swift