Transform ESRGAN Pytorch to TensorRT and Inference Transform Process 1. Pytorch 2 ONNX python torch2onnx.py --input esrgan.pth --output esrgan.onnx --fp16 ONNX 2 TensorRT python onnx2trt.py --input esrgan.onnx --output esrgan.trt --fp16 Test inference with TensorRT ESRGAN python test_trt.py --model esrgan.trt --image input.png --output output.png