From 871bd92329e5bffa25a54e09fa9de3bd1b47b964 Mon Sep 17 00:00:00 2001
From: yunyaoXYY <109218879+yunyaoXYY@users.noreply.github.com>
Date: Thu, 22 Sep 2022 19:27:25 +0800
Subject: [PATCH] Add PaddleInference-GPU backend for OCR Rec model (#271)

* Add PaddleInference GPU backend for Rec model

* Improve the order of deafult backends
---
 fastdeploy/vision/ocr/ppocr/classifier.cc | 2 +-
 fastdeploy/vision/ocr/ppocr/recognizer.cc | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/fastdeploy/vision/ocr/ppocr/classifier.cc b/fastdeploy/vision/ocr/ppocr/classifier.cc
index d0d9d9016..435f18444 100644
--- a/fastdeploy/vision/ocr/ppocr/classifier.cc
+++ b/fastdeploy/vision/ocr/ppocr/classifier.cc
@@ -31,7 +31,7 @@ Classifier::Classifier(const std::string& model_file,
     valid_gpu_backends = {Backend::ORT, Backend::TRT};  // 指定可用的GPU后端
   } else {
     valid_cpu_backends = {Backend::PDINFER, Backend::ORT, Backend::OPENVINO};
-    valid_gpu_backends = {Backend::PDINFER, Backend::TRT, Backend::ORT};
+    valid_gpu_backends = {Backend::PDINFER, Backend::ORT, Backend::TRT};
   }
   runtime_option = custom_option;
   runtime_option.model_format = model_format;
diff --git a/fastdeploy/vision/ocr/ppocr/recognizer.cc b/fastdeploy/vision/ocr/ppocr/recognizer.cc
index 84b90afee..8202ece6b 100644
--- a/fastdeploy/vision/ocr/ppocr/recognizer.cc
+++ b/fastdeploy/vision/ocr/ppocr/recognizer.cc
@@ -48,9 +48,8 @@ Recognizer::Recognizer(const std::string& model_file,
                           Backend::OPENVINO};  // 指定可用的CPU后端
     valid_gpu_backends = {Backend::ORT, Backend::TRT};  // 指定可用的GPU后端
   } else {
-    // NOTE:此模型暂不支持paddle-inference-Gpu推理
-    valid_cpu_backends = {Backend::ORT, Backend::PDINFER, Backend::OPENVINO};
-    valid_gpu_backends = {Backend::ORT, Backend::TRT};
+    valid_cpu_backends = {Backend::PDINFER, Backend::ORT, Backend::OPENVINO};
+    valid_gpu_backends = {Backend::PDINFER, Backend::ORT, Backend::TRT};
   }
 
   runtime_option = custom_option;