tensorflow / tensorflow/models

Depth Multiplier becomes 0, on conversion to TFLITE

Open
#5,767 10 comments 0 reactions 1 assignee View on GitHub

Nobody has claimed this yet.

comp:lite models:research
Dominant language
Python
Stars
77.7k
Forks
44.8k
PR merge metrics
No merged PRs in 30d

Description

MODEL: DeeplabV3+ MobilenetV2 Pretrained with Pascal VOC 2012
Tensorflow Version: 1.12.0
TFLITE Version: 1.12.0
Bazel Version: 0.17.2
OS version: Ubuntu 16.04
GPU Version: GTX 1080 Ti
Command to reproduce:

bazel run //tensorflow/lite/toco:toco --
--input_file=frozen_inference_graph.pb
--output_file=converted.tflite
--input_format=TENSORFLOW_GRAPHDEF
--output_format=TFLITE
--input_arrays=ImageTensor
--output_arrays=SemanticPredictions
--input_shapes=1,513,513,3
--allow_custom_ops
--inference_type=FLOAT
--inference_input_type=QUANTIZED_UINT8

(or)

CUDA_VISIBLE_DEVICES="0"
tflite_convert
--output_file=converted.tflite
--graph_def_file=frozen_inference_graph.pb
--inference_type=FLOAT
--inference_input_type=QUANTIZED_UINT8
--input_arrays=ImageTensor
--output_arrays=SemanticPredictions
--input_shapes=1,513,513,3
--mean_values=513
--std_dev_values=513

Benchmarking Command:
bazel run -c opt tensorflow/lite/tools/benchmark:benchmark_model -- --graph='realpath converted.tflite'

Conversion of pre-trained models provided on the model zoo to TFLITE, on benchmarking and running on Android provided issues on depth multiplier value getting converted to 0.

Stack Overflow Issue: https://stackoverflow.com/questions/53228969/unable-to-test-and-deploy-a-deeplabv3-mobilenetv2-tensorflow-lite-segmentation-m

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.