apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 01:22:59 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bug Fix] Fix bugs when deploy quantized YOLOv5/v6/v7 model. (#729)

Browse Source 
* Improve the usage of fastdeploy tools

* Fix quantized YOLOv5,v6 and v7 model deployment
This commit is contained in:
yunyaoXYY
2022-11-28 19:19:37 +08:00
committed by GitHub
parent 941057888a
commit dc2dad62a4
14 changed files with 12 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/vision/detection/yolov7/quantize/cpp/README.md
View File

@@ -28,7 +28,7 @@ tar -xvf yolov7_quant.tar
wget https://gitee.com/paddlepaddle/PaddleDetection/raw/release/2.4/demo/000000014439.jpg
# 在CPU上使用Paddle-Inference推理量化模型
# 在CPU上使用ONNX Runtime推理量化模型
./infer_demo yolov7_quant 000000014439.jpg 0
# 在GPU上使用TensorRT推理量化模型
./infer_demo yolov7_quant 000000014439.jpg 1

2
examples/vision/detection/yolov7/quantize/cpp/infer.cc
View File

@@ -64,7 +64,7 @@ int main(int argc, char* argv[]) {
if (flag == 0) {
option.UseCpu();
option.UsePaddleBackend();
option.UseOrtBackend();
} else if (flag == 1) {
option.UseGpu();
option.UseTrtBackend();