форма вывода mlmodel отличается от входной - PullRequest
0 голосов
/ 19 июня 2020

Это моя модель spe c описание

input {
  name: "input_2"
  type {
    imageType {
      width: 540
      height: 540
      colorSpace: RGB
    }
  }
}
output {
  name: "Identity"
  type {
    multiArrayType {
      dataType: FLOAT32
    }
  }
}
metadata {
  userDefined {
    key: "coremltoolsVersion"
    value: "3.4"
  }
}

но в Ma c os форма результата (1, 768, 768, 3) ... я пытаюсь преобразовать результат для изображения с помощью coreMLHelpers (https://github.com/hollance/CoreMLHelpers), но возникает ошибка (форма отличается), как я могу решить эту проблему

Более подробная информация о моем коде находится по этой ссылке: mlmodel's форма вывода пуста. Почему фигура пуста?

computer ::: windows10 / tensorflow и -gpu 2.2 / tfcoreml 1.1

...