TOCO - Размеры должны совпадать (оператор Concat) - PullRequest
Новая
       21

TOCO - Размеры должны совпадать (оператор Concat)

0 голосов
/ 26 августа 2018

Я преобразую модель семантической сегментации в 8-битный Quantised TFLite с использованием TOCO, однако получаю следующую ошибку. 

tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc:116] Check failed: dim_x == dim_y (3 vs. 13)Dimensions must match
    Abort trap: 6

Я использую оператор concat TF несколько раз, каждый раз по оси 3, так что размеры concat ([1,20,40,13], [1,20,40,3]) = [1,20,40 , 16], где, очевидно, и возникает ошибка, но вызываемая функция 

bazel build tensorflow/contrib/lite/toco:toco && \
./bazel-bin/tensorflow/contrib/lite/toco/toco \
    --input_file=/path/to/MyModel/log/quantised/optimised_graph.pb \
    --output_file=/path/to/MyModel/log/quantised/quantised_graph.tflite \
    --input_format="TENSORFLOW_GRAPHDEF" \
    --output_format="TFLITE" \
    --inference_type="QUANTIZED_UINT8" \
    --inference_input_type="QUANTIZED_UINT8" \
    --allow_custom_ops=true \
    --drop_control_dependency=false \
    --allow_nudging_weights_to_use_fast_gemm_kernel=true \
    --quantize_weights=false \
    --input_arrays="MyModel/downsampling_block_1_conv/Conv2D","MyModel/downsampling_block_1_pool/MaxPool" \
    --output_arrays="MyModel/Softmax" \
    --input_shapes="1,360,480,3":"1,360,480,3"

Состояния, чтобы разрешить пользовательские операции, которые работают для определенных операций, которые я использую, однако здесь это не работает. От Googling эта ошибка считается еще не реализованной ошибкой типа, но, безусловно, разрешение пользовательских операций должно исправить это?

Наконец, при запуске этой функции создается выходной файл, однако это 0-байтовый пустой файл

...