Может ли модель TensorFlow MTCNN преобразовываться в формат TensorFlow Lite? - PullRequest
Новая
       10

Может ли модель TensorFlow MTCNN преобразовываться в формат TensorFlow Lite?

1 голос
/ 25 марта 2019

Я пытаюсь преобразовать модель MTCNN (https://github.com/blaueck/tf-mtcnn/blob/master/mtcnn.pb) из файла .pb в .tflite и получить проблемы с формами ввода и вывода. Исходная форма ввода -? X? X3, а форма вывода - Nx4, где N - количество обнаруженных лиц.

Я попытался установить входную форму [Нет, Нет, 3] и получить ошибку "Нет поддерживается только в 1-м измерении" . Затем я установил [500, 500, 3] и получил другую ошибку "Проверка не пройдена: пакет == 1 (500 против 1)" . Затем я установил фигуру на [1, 500, 500, 3] и получил "ValueError: форму тензорного 'input' нельзя изменить с (?,?, 3) на [1, 500, 500 , 3]. Фигуры должны быть одинакового ранга, но имеют размеры 3 и 4 ".

UPD: я преобразовал исходную модель кафе из формы ввода [Нет, Нет, 3] в [500, 500, 3], но это не решает проблему.

Я хочу преобразовать эту модель в формат .tflite. Могу ли я на самом деле это сделать?

1 Ответ

0 голосов
/ 15 апреля 2019

пошаговый итог:

использовать официальный репозиторий dr.zhang mtcnn, реализованный caffe: https://github.com/kpzhang93/MTCNN_face_detection_alignment

после этого конвертировать его в модель тензорного потока по ссылке ниже: https://github.com/ethereon/caffe-tensorflow

(даже если вы можете предварительно преобразовать модели, такие как нижеприведенное репо :) https://github.com/AITTSMD/MTCNN-Tensorflow https://github.com/wangbm/MTCNN-Tensorflow (но я настоятельно рекомендую вам сделать это самостоятельно)

наконец, вы можете конвертировать модель TF в TF Lite из документации Google: https://www.tensorflow.org/lite/convert

...