yolo-tensorrt
yolo-tensorrt copied to clipboard
enazoe
int8 engine generation failed
Hi, highly appreciate for your work.
Device: AGX Xavier OS: Ubuntu 18.04.6 LTS Jetpack 4.6, CUDA 10.2, CUDNN 8.2.1, TensorRT 8.0.1.6
We have tried FP32 FP16 for yolov5-6.0 n/s/l/m/x and yolov3 also; this project works perfectly for these mode. But int8 mode does not work, we get this error when we do int8 mode for yolov5-6.0 n/s/m/l/x, yolv3 also
ERROR: 2: [reformatRunner.cpp::onShapeChangeNONCONST::104] Error Code 2: Internal Error (Assertion mCombinedScalesSize >= channelDst failed.) yolo-trt: /home/nvidia/Projects/yolo-tensorrt/modules/yolo.cpp:488: void Yolo::createYOLOEngine(nvinfer1::DataType, Int8EntropyCalibrator*): Assertion `m_Engine != nullptr' failed. Aborted (core dumped)
Hi, highly appreciate for your work.
Device: AGX Xavier OS: Ubuntu 18.04.6 LTS Jetpack 4.6, CUDA 10.2, CUDNN 8.2.1, TensorRT 8.0.1.6
We have tried FP32 FP16 for yolov5-6.0 n/s/l/m/x and yolov3 also; this project works perfectly for these mode. But int8 mode does not work, we get this error when we do int8 mode for yolov5-6.0 n/s/m/l/x, yolv3 also
ERROR: 2: [reformatRunner.cpp::onShapeChangeNONCONST::104] Error Code 2: Internal Error (Assertion mCombinedScalesSize >= channelDst failed.) yolo-trt: /home/nvidia/Projects/yolo-tensorrt/modules/yolo.cpp:488: void Yolo::createYOLOEngine(nvinfer1::DataType, Int8EntropyCalibrator*): Assertion `m_Engine != nullptr' failed. Aborted (core dumped)
Please refer to our open source quantization tool ppq, the quantization effect is better than the calibration of tensorrt, if you encounter issues, we can help you solve them. https://github.com/openppl-public/ppq/blob/master/md_doc/deploy_trt_by_OnnxParser.md
