TensorRT and Tensorflow 2
问题 I am trying to speed up the inference of yolov3 TF2 with TensorRT. I am using the TrtGraphConverter function in tensorflow 2. My code is essentially this: from tensorflow.python.compiler.tensorrt import trt_convert as trt tf.keras.backend.set_learning_phase(0) converter = trt.TrtGraphConverter( input_saved_model_dir="./tmp/yolosaved/", precision_mode="FP16", is_dynamic_op=True) converter.convert() saved_model_dir_trt = "./tmp/yolov3.trt" converter.save(saved_model_dir_trt) And this generates