wehub-resource-sync
e904b667c6
Build/Publish Develop Docs / deploy (push) Failing after 1s
PaddleOCR Code Style Check / check-code-style (push) Failing after 1s
PaddleOCR PR Tests GPU / detect-changes (push) Failing after 1s
PaddleOCR PR Tests / detect-changes (push) Failing after 1s
PaddleOCR PR Tests GPU / test-pr-gpu (push) Has been cancelled
PaddleOCR PR Tests / test-pr (push) Has been cancelled
PaddleOCR PR Tests GPU / test-pr-gpu-impl (push) Has been cancelled
PaddleOCR PR Tests / test-pr-python (3.13) (push) Has been cancelled
PaddleOCR PR Tests / test-pr-python (3.8) (push) Has been cancelled
PaddleOCR PR Tests / test-pr-python (3.9) (push) Has been cancelled
55 lines
1.8 KiB
C++
55 lines
1.8 KiB
C++
// Copyright (c) 2025 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 "src/modules/text_recognition/predictor.h"
|
|
|
|
struct TextRecognitionParams {
|
|
absl::optional<std::string> model_name = absl::nullopt;
|
|
absl::optional<std::string> model_dir = absl::nullopt;
|
|
absl::optional<std::string> lang = absl::nullopt;
|
|
absl::optional<std::string> ocr_version = absl::nullopt;
|
|
absl::optional<std::string> vis_font_dir = absl::nullopt;
|
|
absl::optional<std::string> device = absl::nullopt;
|
|
std::string precision = "fp32";
|
|
bool enable_mkldnn = true;
|
|
int mkldnn_cache_capacity = 10;
|
|
int cpu_threads = 8;
|
|
int batch_size = 1;
|
|
absl::optional<std::vector<int>> input_shape = absl::nullopt;
|
|
};
|
|
|
|
class TextRecognition {
|
|
public:
|
|
TextRecognition(
|
|
const TextRecognitionParams ¶ms = TextRecognitionParams());
|
|
|
|
std::vector<std::unique_ptr<BaseCVResult>> Predict(const std::string &input) {
|
|
std::vector<std::string> inputs = {input};
|
|
return Predict(inputs);
|
|
};
|
|
std::vector<std::unique_ptr<BaseCVResult>>
|
|
Predict(const std::vector<std::string> &input);
|
|
|
|
void CreateModel();
|
|
absl::Status CheckParams();
|
|
static TextRecPredictorParams
|
|
ToTextRecognitionModelParams(const TextRecognitionParams &from);
|
|
|
|
private:
|
|
TextRecognitionParams params_;
|
|
std::unique_ptr<BasePredictor> model_infer_;
|
|
};
|