first commit

fastdeploy/vision/common/processors/pad.h

@@ -0,0 +1,50 @@
+// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#pragma once
+
+#include "fastdeploy/vision/common/processors/base.h"
+
+namespace fastdeploy {
+namespace vision {
+
+class Pad : public Processor {
+ public:
+  Pad(int top, int bottom, int left, int right,
+      const std::vector<float>& value) {
+    top_ = top;
+    bottom_ = bottom;
+    left_ = left;
+    right_ = right;
+    value_ = value;
+  }
+  bool CpuRun(Mat* mat);
+#ifdef ENABLE_OPENCV_CUDA
+  bool GpuRun(Mat* mat);
+#endif
+  std::string Name() { return "Pad"; }
+
+  static bool Run(Mat* mat, const int& top, const int& bottom, const int& left,
+                  const int& right, const std::vector<float>& value,
+                  ProcLib lib = ProcLib::OPENCV_CPU);
+
+ private:
+  int top_;
+  int bottom_;
+  int left_;
+  int right_;
+  std::vector<float> value_;
+};
+} // namespace vision
+} // namespace fastdeploy