tkDNN works with trt8!!,need to test int8 and mobilenet,dla_cnet (fps seems to be a bit low 350 on trt8 compared to 396 on trt7)

2021-09-07 22:00:28 +05:30
parent 9fa116ce4a
commit 03473743c4
30 changed files with 43 additions and 16 deletions
@@ -29,7 +29,7 @@ namespace tk { namespace dnn {
    for(int i=0; i<noYolos.size(); i++) {
        std::vector<float> maskTemp,anchorsTemp;
        std::vector<std::string> classNamesTemp;
-        int classes,nms_kind,coords,numTemp;
+        int nms_kind,coords,numTemp;
        float nmsthresh;
        loadYoloInfo(cfg_path,yolosLine[i],maskTemp,anchorsTemp,numTemp,classes,nmsthresh,nms_kind,coords);
        classNamesTemp = darknetReadNames(name_path);