peace195 / tensorflow-lite-YOLOv3

YOLOv3: convert .weights to .tflite format for tensorflow lite. Convert .weights to .pb format for tensorflow serving
106 stars 24 forks source link

RUN error:tflite_convert --saved_model_dir=./saved_model/ --output_file yolo_v3.tflite --saved_model_signature_key='predict' #6

Closed Byronnar closed 4 years ago

Byronnar commented 4 years ago
"TOCO failed. See console for info.\n%s\n%s\n" % (stdout, stderr))

tensorflow.lite.python.convert.ConverterError: TOCO failed. See console for info. 2019-11-25 09:28:17.070371: I tensorflow/lite/toco/graph_transformations/graph_transformations.cc:39] Before Removing unused ops: 1012 operators, 1562 arrays (0 quantized) 2019-11-25 09:28:17.128413: I tensorflow/lite/toco/graph_transformations/graph_transformations.cc:39] Before general graph transformations: 1012 operators, 1562 arrays (0 quantized) 2019-11-25 09:28:18.329709: F tensorflow/lite/toco/graph_transformations/propagate_fixed_sizes.cc:722] Check failed: start_array.data_type == ArrayDataType::kInt32 Range op inputs must be int32. Aborted (core dumped) I can not convert this ...

peace195 commented 4 years ago

@Byronnar What is your env? (tensorflow version?)

Did you use docker?

Byronnar commented 4 years ago

My tensorflow version is 1.13.1 I have not use docker. I will try it

Byronnar commented 4 years ago

It is ok (run in docker) Thank you!