深度学习系列六:将网络迁移到TensorRT7.0平台
概述 关于使用TensorRT对网络进行加速的基本工作在 深度学习系列五:使用TensorRT对网络进行加速 中描述很详细了,这里主要记录下在迁移到7.0时,顺便解决了之前ResizeNearestNeighbor这个层在uff模型中无法直接支持必须外挂plugin的问题。 Keras模型转ONNX模型 之前使用6.0时,采用的方案是Keras模型转uff模型,再创建engine,其实从官方文档可以看出来,目前TensorRT对onnx支持更好,它对onnx支持的层是最多的,这样容易理解,毕竟onnx可以由各种模型转化得到,因此本次使用onnx模型建立engine,转化可以使用 keras-onnx库 ,安装也很简单: pip3 install keras2onnx 安装完成后使用api进行转换,非常简单,当然还必须安装tensorflow或者keras: import keras2onnx import onnx # load keras model model = tf . compat . v1 . keras . models . load_model ( model_path ) # convert to onnx model onnx_model = keras2onnx . convert_keras ( model , model . name , target