diff --git a/src/MaaCore/Config/Miscellaneous/OcrPack.cpp b/src/MaaCore/Config/Miscellaneous/OcrPack.cpp index a8d2c6ffc6..eeb50312f0 100644 --- a/src/MaaCore/Config/Miscellaneous/OcrPack.cpp +++ b/src/MaaCore/Config/Miscellaneous/OcrPack.cpp @@ -27,17 +27,13 @@ static std::filesystem::path prepare_paddle_dir(const std::filesystem::path& dir #endif asst::OcrPack::OcrPack() - : m_ocr_option(std::make_unique()), m_det(nullptr), m_rec(nullptr), m_ocr(nullptr), - m_backend(OcrBackend::ONNXRuntime) -{} + : m_ocr_option(std::make_unique()), m_det(nullptr), m_rec(nullptr), m_ocr(nullptr) +{ + m_ocr_option->UseOrtBackend(); +} asst::OcrPack::~OcrPack() {} -void asst::OcrPack::set_backend_before_load(OcrBackend backend) -{ - m_backend = backend; -} - bool asst::OcrPack::load(const std::filesystem::path& path) { bool use_temp_dir = false; diff --git a/src/MaaCore/Config/Miscellaneous/OcrPack.h b/src/MaaCore/Config/Miscellaneous/OcrPack.h index ab7dcd4c87..5b77486a9b 100644 --- a/src/MaaCore/Config/Miscellaneous/OcrPack.h +++ b/src/MaaCore/Config/Miscellaneous/OcrPack.h @@ -30,16 +30,9 @@ namespace asst { class OcrPack : public AbstractResource { - public: - enum class OcrBackend - { - ONNXRuntime, - }; public: virtual ~OcrPack() override; - void set_backend_before_load(OcrBackend backend); - virtual bool load(const std::filesystem::path& path) override; std::vector recognize(const cv::Mat& image, const TextRectProc& pred = nullptr, @@ -54,8 +47,6 @@ namespace asst std::unique_ptr m_det; std::unique_ptr m_rec; std::unique_ptr m_ocr; - - OcrBackend m_backend; }; class WordOcr final : public SingletonHolder, public OcrPack