美文网首页
TensorRT maskrcnn windows下使用自己的数

TensorRT maskrcnn windows下使用自己的数

作者: 半笔闪 | 来源:发表于2020-09-21 16:21 被阅读0次
  • 环境
    系统: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可看到结果

相关文章

网友评论

      本文标题:TensorRT maskrcnn windows下使用自己的数

      本文链接:https://www.haomeiwen.com/subject/nkwvyktx.html