TensorRT maskrcnn windows下使用自己的數(shù)據(jù)集（二）

  • 環(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)備測試數(shù)據(jù)
    如果要不更改代碼使用TensorRT中maskrcnn的sample，那需要ppm格式的圖片測試，這很簡單，只要準(zhǔn)備好自己的jpg測試數(shù)據(jù)就行，然后用PIL轉(zhuǎn)換一下，如下：
from PIL import Image
image = Image.open("你的測試數(shù)據(jù).jpg")
image.save("input.ppm")

這樣你就可以得到名未input.ppm的測試數(shù)據(jù)了

  • 配置TensorRT Maskrcnn環(huán)境
    1、解壓TensorRT7.0.0.11，進入data/文件夾下，新建maskrcnn文件夾，把input.ppm復(fù)制到maskrcnn下，把之前生成的mrcnn_nchw.uff也復(fù)制到maskrcnn下；
    2、進入到sample/sampleUffMaskrcnn文件夾下，用vs2019打開sln文件，右鍵項目設(shè)置為啟動項目，如果希望生成調(diào)試程序，右鍵項目-》屬性-》鏈接器-》調(diào)試-》生成調(diào)試信息，選擇生成調(diào)試信息（/DEBUG）；
    3、打開sampleUffMaskRCNN.cpp，修改輸入，487行
std::vector<std::string> imageListCandidates = {"input.ppm"};

4、點擊運行，不出意外生成結(jié)果0.ppm，使用PIL轉(zhuǎn)換成jpg可看到結(jié)果

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容