int8-support-

src/libs/inferences/tensorrt/trt_infer.cpp

@@ -190,8 +190,11 @@ bool TRTInference::BuildEngine(YAML::Node& configNode){
                 configNode["inputWidth"].as<int>(), configNode["inputHeight"].as<int>(), 
                 inputMean, inputStd, configNode["useBGR"].as<bool>(), configNode["fixAspectRatio"].as<bool>()
             );
-            // builder->setInt8Calibrator(calibrator);
+            #ifdef TRT_8
             builderConfig->setInt8Calibrator(calibrator);
+            #else
+            builder->setInt8Calibrator(calibrator);
+            #endif
         }
         else{
             LOG_ERROR << "Not imgs for calib int8. " << configNode["calibImgPathFile"].as<std::string>();

src/libs/inferences/tensorrt/trt_yolov5_infer.cpp

@@ -372,7 +372,11 @@ bool YoloV5TRTInference::BuildEngine(YAML::Node& configNode){
                         configNode["calibTableCache"].as<std::string>(), 
                         maxBatchSize, inputWidth, inputHeight, inputMean, inputStd, 
                         configNode["useBGR"].as<bool>(), configNode["fixAspectRatio"].as<bool>());
+        #ifdef TRT_8
         config->setInt8Calibrator(calibrator);
+        #else
+        builder->setInt8Calibrator(calibrator);
+        #endif
         // */
     //    return false;
     }