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