модель caffemodel to coreml: невозможно определить форму для ошибки ввода и сегментации после добавления input_shape - PullRequest
Новая
       24

модель caffemodel to coreml: невозможно определить форму для ошибки ввода и сегментации после добавления input_shape

0 голосов
/ 01 декабря 2018

Я пытаюсь преобразовать caffemodel в модель coreml, используя coremltools.Мой код ниже: 

import coremltools

model_path = '../models/'

coreml_model = coremltools.converters.caffe.convert((model_path + 'snapshots/itracker_iter_92000.caffemodel', model_path + 'itracker_deploy.prototxt', {'image_left': model_path + 'mean_images/mean_left_224.binaryproto', 'image_right': model_path + 'mean_images/mean_right_224.binaryproto', 'image_face': model_path + 'mean_images/mean_face_224.binaryproto'}), image_input_names=['image_left', 'image_right', 'image_face'])
coreml_model.save('eyes.mlmodel')

Проблема в том, что я получаю следующее сообщение об ошибке:

RuntimeError: Невозможно определить форму для входных данных 'facegrid'.

В моем deploy.prototxt у меня есть следующие строки: 

input: "facegrid"
input_dim: 256
input_dim: 625
input_dim: 1
input_dim: 1

После поиска некоторых других вопросов StackOverflow я следовал за этим: CoreСбой преобразования модели ML: «Невозможно определить имя и размеры входных данных» , и я добавил следующие строки в файл: 

input: "facegrid"
input_dim: 256
input_dim: 625
input_dim: 1
input_dim: 1
input_shape {
  dim: 1
  dim: 1
  dim: 25
  dim: 25
}

Но теперь я получаю только ошибку Сегментация: 11 .Может ли кто-нибудь помочь мне?Дерево каталогов проекта таково:

Дерево каталогов проекта

Спасибо!

...