- 环境
系统:windows10
工具:CUDA10.0.130、cudnn7.6.3.30、TensorRT7.0.0.11
IDE:Visual Studio2019 - 安装cuda,cudnn,vs2019
参考https://blog.csdn.net/qq_37296487/article/details/83028394 - 准备测试数据
如果要不更改代码使用TensorRT中maskrcnn的sample,那需要ppm格式的图片测试,这很简单,只要准备好自己的jpg测试数据就行,然后用PIL转换一下,如下:
from PIL import Image
image = Image.open("你的测试数据.jpg")
image.save("input.ppm")
这样你就可以得到名未input.ppm的测试数据了
- 配置TensorRT Maskrcnn环境
1、解压TensorRT7.0.0.11,进入data/文件夹下,新建maskrcnn文件夹,把input.ppm复制到maskrcnn下,把之前生成的mrcnn_nchw.uff也复制到maskrcnn下;
2、进入到sample/sampleUffMaskrcnn文件夹下,用vs2019打开sln文件,右键项目设置为启动项目,如果希望生成调试程序,右键项目-》属性-》链接器-》调试-》生成调试信息,选择生成调试信息(/DEBUG);
3、打开sampleUffMaskRCNN.cpp,修改输入,487行
std::vector<std::string> imageListCandidates = {"input.ppm"};
4、点击运行,不出意外生成结果0.ppm,使用PIL转换成jpg可看到结果