Modify PPMatting backend and docs

csrc/fastdeploy/vision/matting/ppmatting/ppmatting.cc

@@ -27,7 +27,7 @@ PPMatting::PPMatting(const std::string& model_file,
                      const Frontend& model_format) {
   config_file_ = config_file;
   valid_cpu_backends = {Backend::PDINFER, Backend::ORT};
-  valid_gpu_backends = {Backend::PDINFER, Backend::ORT, Backend::TRT};
+  valid_gpu_backends = {Backend::PDINFER, Backend::TRT};
   runtime_option = custom_option;
   runtime_option.model_format = model_format;
   runtime_option.model_file = model_file;

csrc/fastdeploy/vision/matting/ppmatting/ppmatting.h

@@ -27,7 +27,7 @@ class FASTDEPLOY_DECL PPMatting : public FastDeployModel {
             const RuntimeOption& custom_option = RuntimeOption(),
             const Frontend& model_format = Frontend::PADDLE);
 
-  std::string ModelName() const { return "PaddleMat"; }
+  std::string ModelName() const { return "PaddleMatting"; }
 
   virtual bool Predict(cv::Mat* im, MattingResult* result);
 

examples/vision/matting/ppmatting/cpp/README.md

@@ -27,8 +27,6 @@ wget https://bj.bcebos.com/paddlehub/fastdeploy/matting_bgr.jpg
 
 # CPU推理
 ./infer_demo PP-Matting-512 matting_input.jpg matting_bgr.jpg 0
-# GPU推理 (TODO: ORT-GPU 推理会报错)
-./infer_demo PP-Matting-512 matting_input.jpg matting_bgr.jpg 1
 # GPU上TensorRT推理
 ./infer_demo PP-Matting-512 matting_input.jpg matting_bgr.jpg 2
 ```

examples/vision/matting/ppmatting/cpp/infer.cc

@@ -50,39 +50,6 @@ void CpuInfer(const std::string& model_dir, const std::string& image_file,
             << std::endl;
 }
 
-void GpuInfer(const std::string& model_dir, const std::string& image_file,
-              const std::string& background_file) {
-  auto model_file = model_dir + sep + "model.pdmodel";
-  auto params_file = model_dir + sep + "model.pdiparams";
-  auto config_file = model_dir + sep + "deploy.yaml";
-
-  auto option = fastdeploy::RuntimeOption();
-  option.UseGpu();
-  auto model = fastdeploy::vision::matting::PPMatting(model_file, params_file,
-                                                      config_file, option);
-  if (!model.Initialized()) {
-    std::cerr << "Failed to initialize." << std::endl;
-    return;
-  }
-
-  auto im = cv::imread(image_file);
-  auto im_bak = im.clone();
-  cv::Mat bg = cv::imread(background_file);
-  fastdeploy::vision::MattingResult res;
-  if (!model.Predict(&im, &res)) {
-    std::cerr << "Failed to predict." << std::endl;
-    return;
-  }
-  auto vis_im = fastdeploy::vision::Visualize::VisMattingAlpha(im_bak, res);
-  auto vis_im_with_bg =
-      fastdeploy::vision::Visualize::SwapBackgroundMatting(im_bak, bg, res);
-  cv::imwrite("visualized_result.jpg", vis_im_with_bg);
-  cv::imwrite("visualized_result_fg.jpg", vis_im);
-  std::cout << "Visualized result save in ./visualized_result_replaced_bg.jpg "
-               "and ./visualized_result_fg.jpgg"
-            << std::endl;
-}
-
 void TrtInfer(const std::string& model_dir, const std::string& image_file,
               const std::string& background_file) {
   auto model_file = model_dir + sep + "model.pdmodel";
@@ -131,8 +98,6 @@ int main(int argc, char* argv[]) {
   }
   if (std::atoi(argv[4]) == 0) {
     CpuInfer(argv[1], argv[2], argv[3]);
-  } else if (std::atoi(argv[4]) == 1) {
-    GpuInfer(argv[1], argv[2], argv[3]);
   } else if (std::atoi(argv[4]) == 2) {
     TrtInfer(argv[1], argv[2], argv[3]);
   }

examples/vision/matting/ppmatting/python/README.md

@@ -19,8 +19,6 @@ wget https://bj.bcebos.com/paddlehub/fastdeploy/matting_input.jpg
 wget https://bj.bcebos.com/paddlehub/fastdeploy/matting_bgr.jpg
 # CPU推理
 python infer.py --model PP-Matting-512 --image matting_input.jpg --bg matting_bgr.jpg --device cpu
-# GPU推理 (TODO: ORT-GPU 推理会报错)
-python infer.py --model PP-Matting-512 --image matting_input.jpg --bg matting_bgr.jpg --device gpu
 # GPU上使用TensorRT推理 （注意：TensorRT推理第一次运行，有序列化模型的操作，有一定耗时，需要耐心等待）
 python infer.py --model PP-Matting-512 --image matting_input.jpg --bg matting_bgr.jpg --device gpu --use_trt True
 ```