TensorRT5輸出engine網路層

2021-09-02 18:21:45 字數 1662 閱讀 4521

由於tensorrt5的api相對於tensorrt4有了不少改變,特別是刪除了config類,無法列印轉換後生成的trt_engine網路層。通過依次讀取network每個layer並判斷layer類別,輸出layer的設定屬性,可以達到更好的檢視效果。

**如下:

const std::string kshowlayertype[24] = ;

const std::string kshowactivationtype[3] = ;

const std::string kshowpoolingtype[3] = ;

const std::string kshowscaletype[3] = ;

const std::string kshowelementwisetype[7] = ;

const std::string kshowunarytype[6] = ;

const std::string kshowreducetype[5] = ;

const std::string kshowmatrixoptype[3] = ;

const std::string kshowdatatype[4] = ;

void printlayerinfo(nvinfer1::inetworkdefinition *network)

else

std::cout << showstring << std::endl;

break;

}case 1:

else

std::cout << showstring << std::endl;

break;

}case 2:

case 4:

case 5:

case 7:

else

std::cout << showstring << std::endl;

break;

}case 8:

case 9:

case 11:

case 12:

case 14:

;if(!(order[0]==0 && order[1]==1 && order[2]==2))

//second transpose order

porder = shufflelayer->getsecondtranspose().order;

order[0] = porder[0];

order[1] = porder[1];

order[2] = porder[2];

if(order[0]!=0 || order[1] !=1 || order[2]!=2)

//reshape dims

nvinfer1::dims rdims = shufflelayer->getreshapedimensions();

if(rdims.nbdims==3)

case 16:

case 17:

case 18:

case 19:

case 20:

case 22:

case 23:

default:

std::cout << i << "--nontype layer" << std::endl;

break;}}

}

tensorrt5版本 安裝TensorRT

解壓安裝包 version 6.0.1.8 os ubuntu 16.04 arch uname m cuda cuda 10.2 cudnn cudnn7.6 tar xzvf tensorrt gnu.tar.gz 解壓得到tensorrt 6.0.1.8的資料夾,將裡邊的lib絕對路徑新增到環...

perl學習 5 輸入和輸出

行輸入操作在到達檔案的結尾時將返回 undef 在while迴圈的條件中不能使用chomp while defined line 在迴圈的內部,我們將看到每一行,一行接著一行,perl提供簡寫 while while defined 兩者是等同的 在進行深入討論前,我們要澄清一些事 這種簡寫只在特定...

5 流的輸入與輸出

c 定義的標準流主要包括 cin cout cerr clog cin 是由終端輸入資料,預設為鍵盤 cout 為向終端輸出資料,預設為螢幕 cerr 是向終端輸出標準錯誤資訊,預設為螢幕 clog 為 cerr 的緩衝形式,預設為螢幕。用部落格鏈結 中的例子,使用 cin 和 cout 計算兩個日...