Сжать график tflite от 8 МБ до 2 МБ - PullRequest
Новая
       18

Сжать график tflite от 8 МБ до 2 МБ

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

У меня есть следующий замороженный вывод логики deeplabv3_mnv2_pascal_tain . Я получил этот график с сайта tenorflows Размер этого графа составляет 8,8 Мб. Моя задача - преобразовать этот граф в формат tflite таким образом, чтобы размер графа tflite составлял от 2 до 3 МБ.

Я использую следующий код для преобразования deeplabv3_mnv2_pascal_tain графика в tflite: 

$ bazel-bin/tensorflow/lite/toco/toco \
   --input_file=deeplabv3_mnv2_pascal_tain.pb \
   --output_file=output2/test.tflite \
   --inference_input_type=QUANTIZED_UINT8  \
   --inference_type=FLOAT \
   --input_arrays=ImageTensor  \
   --output_arrays=SemanticPredictions  \
   --input_shapes=1,513,513,3 \
   --mean_values=128 \
   --std_dev_values=128

Тем не менее, выходной tflite-graph, который я получаю, имеет размер 8,5 МБ. Вы можете увидеть выходной график здесь .

Как мне сжать этот tflite-граф так, чтобы я получил график 2,7 МБ

Чтобы быть более понятным, я хочу получить этот тфлитный граф, сжав deeplabv3_mnv2_pascal_tain.pb file

1 Ответ

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

Мне не нужно конвертировать 8 МБ граф в 3 МБ, я обнаружил, что 3 МБ граф уже доступен здесь http://download.tensorflow.org/models/deeplabv3_mnv2_dm05_pascal_trainaug_2018_10_01.tar.gz

...