fz-mirrors/MaaAssistantArknights
0
0
Fork 0
mirror of https://github.com/MaaAssistantArknights/MaaAssistantArknights.git synced 2026-07-17 10:00:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support GPU OCR

Browse Source
This commit is contained in:
MistEO
2023-10-11 17:44:18 +08:00
parent e32c580c76
commit 4f1bf344c4
4 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/MaaCore/Assistant.cpp
View File

@@ -5,6 +5,7 @@
#include <meojson/json.hpp>
#include "Config/GeneralConfig.h"
#include "Config/Miscellaneous/OcrPack.h"
#include "Config/ResourceLoader.h"
#include "Controller/Controller.h"
#include "Status.h"
@@ -34,6 +35,24 @@ using namespace asst;
bool ::AsstExtAPI::set_static_option(StaticOptionKey key, const std::string& value)
{
Log.info(__FUNCTION__, "| key", static_cast<int>(key), "value", value);
switch (key) {
case StaticOptionKey::CpuOCR: {
WordOcr::get_instance().use_cpu();
CharOcr::get_instance().use_cpu();
return true;
} break;
case StaticOptionKey::GpuOCR: {
int device_id = std::stoi(value);
WordOcr::get_instance().use_gpu(device_id);
CharOcr::get_instance().use_gpu(device_id);
return true;
} break;
default:
Log.error(__FUNCTION__, "| unknown key:", static_cast<int>(key));
break;
}
return false;
}