模型实战(18)之C++ - tensorRT部署GAN模型实现人脸超分辨重建

模型实战(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
 

猜你喜欢

转载自blog.csdn.net/yohnyang/article/details/135142163