openppl-public / ppq

PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
Apache License 2.0
1.44k stars 223 forks source link

yolov5量化,在ppq/samples/yolo中运行02_Quantization.py文件出现警告和报错,输出模型只有onnx没有engine文件 #550

Open zjhleaning opened 2 months ago

zjhleaning commented 2 months ago

[Info] You are exporting PPQ Graph to TensorRT(Onnx + Json). Please Compile the TensorRT INT8 engine manually:

from ppq.utils.TensorRTUtil import build_engine build_engine(onnx_file='Quantized.onnx', int8_scale_file='Quantized.json', engine_file='Quantized.engine', int8=True)

Traceback (most recent call last): File "E:\BaiduNetdiskDownload\Data\ppq-master\ppq\samples\Yolo\02_Quantization.py", line 51, in config_save_to=OUTPUT_PATH.join('/INT8.json')) File "E:\BaiduNetdiskDownload\Data\ppq-master\ppq\api\interface.py", line 590, in export_ppq_graph exporter.export(file_path=graph_save_to, config_path=config_save_to, graph=graph, **kwargs) File "E:\BaiduNetdiskDownload\Data\ppq-master\ppq\parser\tensorRT.py", line 123, in export self.export_quantization_config(config_path, graph) File "E:\BaiduNetdiskDownload\Data\ppq-master\ppq\parser\tensorRT.py", line 90, in export_quantization_config with open(config_path, "w") as json_file: FileNotFoundError: [Errno 2] No such file or directory: '/E:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnxIE:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnxNE:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnxTE:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnx8E:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnx.E:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnxjE:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnxsE:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnxoE:/BaiduNetdiskDownload/Data/ppq-master/Output/DET(ppq).onnxn'

Process finished with exit code 1

ZhangZhiPku commented 2 months ago

tensorRT engine 生成失败了,具体是为什么你可能就需要看看了