- 環(huán)境
系統(tǒng):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 - 準(zhǔn)備測(cè)試數(shù)據(jù)
如果要不更改代碼使用TensorRT中maskrcnn的sample,那需要ppm格式的圖片測(cè)試,這很簡(jiǎn)單穗慕,只要準(zhǔn)備好自己的jpg測(cè)試數(shù)據(jù)就行,然后用PIL轉(zhuǎn)換一下妻导,如下:
from PIL import Image
image = Image.open("你的測(cè)試數(shù)據(jù).jpg")
image.save("input.ppm")
這樣你就可以得到名未input.ppm的測(cè)試數(shù)據(jù)了
- 配置TensorRT Maskrcnn環(huán)境
1逛绵、解壓TensorRT7.0.0.11怀各,進(jìn)入data/文件夾下,新建maskrcnn文件夾术浪,把input.ppm復(fù)制到maskrcnn下瓢对,把之前生成的mrcnn_nchw.uff也復(fù)制到maskrcnn下;
2胰苏、進(jìn)入到sample/sampleUffMaskrcnn文件夾下硕蛹,用vs2019打開(kāi)sln文件,右鍵項(xiàng)目設(shè)置為啟動(dòng)項(xiàng)目硕并,如果希望生成調(diào)試程序法焰,右鍵項(xiàng)目-》屬性-》鏈接器-》調(diào)試-》生成調(diào)試信息,選擇生成調(diào)試信息(/DEBUG)倔毙;
3埃仪、打開(kāi)sampleUffMaskRCNN.cpp,修改輸入陕赃,487行
std::vector<std::string> imageListCandidates = {"input.ppm"};
4卵蛉、點(diǎn)擊運(yùn)行,不出意外生成結(jié)果0.ppm么库,使用PIL轉(zhuǎn)換成jpg可看到結(jié)果