登录 注册
当前位置:主页 > 资源下载 > 49 > CenterNet-TensorRT是CenterNet的C++实现,采用TensorRT和CUDA技术

CenterNet-TensorRT是CenterNet的C++实现,采用TensorRT和CUDA技术

  • 更新:2024-12-11 23:57:02
  • 大小:6.87MB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

这是使用TensorRT和CUDA的CenterNet的C ++实现。 感谢的正式实施 ! 依存关系: Ubuntu 16.04 PyTorch 1.2.0(用于与Tenson RT5在Jetson Tx2中的兼容性) CUDA 10.0 [必需] TensorRT-7.0.0.11(用于CUDA10.0)[必需] CUDNN(对于CUDA10.0,可能不使用)[必需] libtorch(cpu版本的Torch c ++ lib,gpu版本可能与环境冲突)[可选] gtest(Google C ++测试框架)[可选] 注意 TensorRT库必须与已安装的CUDA和CUDNN一致 TensorRT 5不支持动态形状 TensorRT 7.0.x不直接支持动态形状的Int8校准 TensorRT 7.1.x支持动态形状的Int8校准 TensorRT的插件: MyUpsam