由於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 計算兩個日...