2020年5月21日
TensorRT推理加速-基于Tensorflow(keras)的uff格式模型(文件准备)
摘要： 一、引子//Windows tf（keras）训练好了模型，想要用Nvidia-TensorRT来重构训练好的模型为TRT推理引擎加快推理的速度。 二、准备文件 1、训练好模型以后（keras）可以通过以下方式保存keras模型为h5文件 tf.keras.models.save_model(mod 阅读全文
posted @ 2020-05-21 18:51 卡冈图雅blackhole 阅读(4061) 评论(0) 推荐(0) 编辑