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

tflite_example.py error when running inferences #12

Open SirTapir opened 4 years ago

SirTapir commented 4 years ago

Hello, I tried to run tflite_example.py on multiple different images. Some runs smoothly, other crashed with different error code. The images can be found here

scream.jpg

Traceback (most recent call last):
  File "tflite_example.py", line 160, in <module>
    draw_boxes(boxes, classes, scores, img, class_names, (height, width), True)
  File "tflite_example.py", line 106, in draw_boxes
    for box, score, cls in zip(boxes, scores, classes):
TypeError: zip argument #1 must support iteration

girl.png

Traceback (most recent call last):
  File "tflite_example.py", line 153, in <module>
    interpreter.set_tensor(input_details[0]['index'], np.expand_dims(img_resized, 0))
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/interpreter.py", line 197, in set_tensor
    self._interpreter.SetTensor(tensor_index, value)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/lite/python/interpreter_wrapper/tensorflow_wrap_interpreter_wrapper.py", line 136, in SetTensor
    return _tensorflow_wrap_interpreter_wrapper.InterpreterWrapper_SetTensor(self, i, value)
ValueError: Cannot set tensor: Dimension mismatch. Got 4 but expected 3 for dimension 3 of input 103.

kite.jpg and person.jpg runs fine