TCT模型部署步骤(1:准备)

本文详细介绍了如何在Windows环境中安装Python 3.8.8、CUDA 11.1,并通过CMake配置TensorRT、OpenCV等依赖,以及如何在TCTDetect项目中部署模型和调试。涉及工具安装、库路径设置和关键步骤的修改。
摘要由CSDN通过智能技术生成

1. 工具及库依赖

1.1. 需要安装的工具

  1. Visual Studio
  2. python3.8
  • 资源路径: D:\安装包\python-3.8.8-amd64
  1. CUDA11.1
  • 默认安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
  • 压缩包中有一个cudnn-11.2-windows-x64-v8.1.1.33,要将该路径bin文件夹中的7个文件,拷贝到上面安装路径bin文件夹中;
  • 资源路径:E:\开发工具资源\TCT模型部署相关\cuda11.1.7z

1.2. 依赖的包

  • 使用路径:E:\Dev\TCTDetect\libs
  • 资源路径:E:\开发工具资源\TCT模型部署相关\libs.7z

1.3. 模型文件

用生成工具在本机生成331_best.engine、negative_yolo.engine两个模型;

  • 使用路径:E:\Dev\TCTDetect\models\convert
  • 资源路径:E:\开发工具资源\TCT模型部署相关\convert

1.4. 调试环境模拟

  • 使用路径:E:\Dev\TCTDetect\Projects
  • 资源路径:E:\开发工具资源\TCT模型部署相关\Projects.7z

2. 修改CMake文件

  • 注: Torch相关的要注释掉
1. Classifier
set(OpenCV_DIR "E:/Dev/TCTDetect/libs/opencv452/build")
set(TENSORRT_DIR "E:/Dev/TCTDetect/libs/TensorRT-7.2.3.4")
set(CUDA_DIR "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.1")
2. ObjectDetect
set(TENSORRT_DIR "E:/Dev/TCTDetect/libs/TensorRT-7.2.3.4")
set(CUDA_DIR "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.1")
set(TCTINFER_DIR "E:/Dev/TCTDetect/libs/tctinfer")
3. PreProcess
set(CUDA_DIR "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.1")
set(OpenCV_DIR "E:/Dev/TCTDetect/libs/opencv452/build")
4. redis
set(REDIS_INCLUDE "E:/Dev/TCTDetect/libs/redis-3.0/deps/hiredis")
set(REDIS_LIB "E:/Dev/TCTDetect/libs/redis-3.0/msvs/x64/Release")
5. RNN
set(TENSORRT_DIR "E:/Dev/TCTDetect/libs/TensorRT-7.2.3.4")
set(CUDA_DIR "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.1")
set(LOG_DIR "E:/Dev/TCTDetect/libs/spdlog-1.x")
6. Slide
set(OpenCV_DIR "E:/Dev/TCTDetect/libs/opencv452/build")
set(OPENSLIDE_DIR "E:/Dev/TCTDetect/libs/openslide-win64-20171122")
set(SDPC_DIR "E:/Dev/TCTDetect/libs/sdpc")

7. 最外层

set(OpenCV_DIR "E:/Dev/TCTDetect/libs/opencv452/build")
set(TENSORRT_DIR "E:/Dev/TCTDetect/libs/TensorRT-7.2.3.4")
set(CUDA_DIR "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.1")
set(LOG_DIR "E:/Dev/TCTDetect/libs/spdlog-1.x")
set(TCTINFER_DIR "E:/Dev/TCTDetect/libs/tctinfer")
set(JSON_DIR "E:/Dev/TCTDetect/libs/json")
set(REDIS_INCLUDE "E:/Dev/TCTDetect/libs/redis-3.0/deps/hiredis")
set(REDIS_LIB "E:/Dev/TCTDetect/libs/redis-3.0/msvs/x64/Release")

3. 用CMake编译源文件

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值