1 Star 0 Fork 0

tom/pytorch_to_tflite

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
run_tflite_model.py 778 Bytes
一键复制 编辑 原始数据 按行查看 历史
kwakhyunmin 提交于 3年前 . first
import numpy as np
import tensorflow as tf
tflite_model_path = 'model.tflite'
# Load the TFLite model and allocate tensors
interpreter = tf.lite.Interpreter(model_path=tflite_model_path)
interpreter.allocate_tensors()
# Get input and output tensors
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()
# Test the model on random input data
input_shape = input_details[0]['shape']
input_data = np.array(np.random.random_sample(input_shape), dtype=np.float32)
interpreter.set_tensor(input_details[0]['index'], input_data)
interpreter.invoke()
# get_tensor() returns a copy of the tensor data
# use tensor() in order to get a pointer to the tensor
output_data = interpreter.get_tensor(output_details[0]['index'])
print(output_data)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/tom_ud/pytorch_to_tflite.git
git@gitee.com:tom_ud/pytorch_to_tflite.git
tom_ud
pytorch_to_tflite
pytorch_to_tflite
main

搜索帮助