模型实战(18)之C++ - tensorRT部署GAN模型实现人脸超分辨重建
- 一个实现人脸超分辨率重建的demo
- 支持StyleGAN: GPEN or GFPGAN
- 通过C++ - tensorrt 快速部署,推理速度每帧 在RTX3090上5.5ms+,RTX3050上10ms+
- 下边是实现效果(图片来源于网络search,如若侵权,联系删除)
- 下边给出实现步骤:
1. 模型转换
- 下载模型至本地
Download the model (BaiduPan pwd: yxt7 or Google Drive) - 通过tensorrt安装路径下的
./bin
文件夹中的trtexec工具编译生成engine