This pull request introduces support for TensorRT-based Real-ESRGAN upscaling in the lite module. The most important changes include the addition of a new test function, updates to the models.h file to include the new model, and the implementation of the Real-ESRGAN class for TensorRT.
TensorRT Real-ESRGAN Integration:
New Test Function:
Added test_tensorrt() function in test_lite_realesrgan.cpp to test TensorRT-based Real-ESRGAN upscaling. (examples/lite/cv/test_lite_realesrgan.cpp, examples/lite/cv/test_lite_realesrgan.cppR23-R46)
Model Header Updates:
Included trt_realesrgan.h in the models.h file. (lite/models.h, lite/models.hR137)
Added typedef for TRTRealESRGAN and nested it within the upscale namespace. (lite/models.h, [1][2]
Real-ESRGAN Implementation:
Implemented the TRTRealESRGAN class with methods for preprocessing, detection, and postprocessing in trt_realesrgan.cpp. (lite/trt/cv/trt_realesrgan.cpp, lite/trt/cv/trt_realesrgan.cppR1-R109)
This pull request introduces support for TensorRT-based Real-ESRGAN upscaling in the
litemodule. The most important changes include the addition of a new test function, updates to themodels.hfile to include the new model, and the implementation of the Real-ESRGAN class for TensorRT.TensorRT Real-ESRGAN Integration:
New Test Function:
test_tensorrt()function intest_lite_realesrgan.cppto test TensorRT-based Real-ESRGAN upscaling. (examples/lite/cv/test_lite_realesrgan.cpp, examples/lite/cv/test_lite_realesrgan.cppR23-R46)Model Header Updates:
trt_realesrgan.hin themodels.hfile. (lite/models.h, lite/models.hR137)TRTRealESRGANand nested it within theupscalenamespace. (lite/models.h, [1] [2]Real-ESRGAN Implementation:
TRTRealESRGANclass with methods for preprocessing, detection, and postprocessing intrt_realesrgan.cpp. (lite/trt/cv/trt_realesrgan.cpp, lite/trt/cv/trt_realesrgan.cppR1-R109)TRTRealESRGANclass in the header filetrt_realesrgan.h. (lite/trt/cv/trt_realesrgan.h, lite/trt/cv/trt_realesrgan.hR1-R27)