Android上的图片字幕

时间:2019-12-29 15:04:08

标签: python android tensorflow

我想使用tflite ML Kit将图像字幕模型嵌入到Android手机中 并在文档中使用:

import tensorflow as tf

interpreter = tf.lite.Interpreter(model_path="my_model.tflite")
interpreter.allocate_tensors()

# Print input shape and type
print(interpreter.get_input_details()[0]['shape'])  # Example: [1 224 224 3]
print(interpreter.get_input_details()[0]['dtype'])  # Example: < class 'numpy.float32' >

# Print output shape and type
print(interpreter.get_output_details()[0]['shape'])  # Example: [1 1000]
print(interpreter.get_output_details()[0]['dtype'])  # Example: < class 'numpy.float32' >

我的输出是一个字符串,我应该写什么作为输出

0 个答案:

没有答案