Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@frogermcs

frogermcs/resize_interpreter.py

Created May 27, 2019 21:28
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save frogermcs/41ba1572340d23e80ac58386c0d91037 to your computer and use it in GitHub Desktop.
Save frogermcs/41ba1572340d23e80ac58386c0d91037 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
resize_interpreter.py
tflite_interpreter.resize_tensor_input(input_details[0]['index'], (32, 224, 224, 3))
tflite_interpreter.resize_tensor_input(output_details[0]['index'], (32, 5))
tflite_interpreter.allocate_tensors()
input_details = tflite_interpreter.get_input_details()
output_details = tflite_interpreter.get_output_details()
print("== Input details ==")
print("shape:", input_details[0]['shape'])
print("\n== Output details ==")
print("shape:", output_details[0]['shape'])
# >> == Input details ==
# >> shape: [ 32 224 224 3]
# >> == Output details ==
# >> shape: [32 5]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment