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

chore: 调整截图保存逻辑

Browse Source
This commit is contained in:
uye
2025-05-18 12:23:26 +08:00
parent 7a65be5fa5
commit 6601c6621c
1 changed files with 53 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
src/MaaCore/Vision/Battle/BattlefieldClassifier.cpp
View File

@@ -99,8 +99,8 @@ BattlefieldClassifier::SkillReadyResult BattlefieldClassifier::skill_ready_analy
SkillReadyResult::Prob prob = softmax(raw_results);
Log.info(__FUNCTION__, "prob:", prob);
// 类别顺序为 c, n, y
size_t class_id = ranges::max_element(prob) - prob.begin();
bool ready = (class_id == 2); // 只有当class_id为2（代表y）时，才认为是ready
int8_t class_id = static_cast<int8_t>(ranges::max_element(prob) - prob.begin());
bool ready = class_id == 2; // 只有当class_id为2（代表y）时，才认为是ready
float score = prob[class_id];
#ifdef ASST_DEBUG
@@ -131,52 +131,70 @@ BattlefieldClassifier::SkillReadyResult BattlefieldClassifier::skill_ready_analy
}
// 为重新训练模型截图
static Point last_base_point = { -1, -1 };
static int last_class = -1; // 记录上一次的分类结果
static auto last_save_time = std::chrono::steady_clock::now();
struct point_state
{
int8_t last_class = -1;
std::chrono::steady_clock::time_point last_save_time;
};
static std::vector<point_state> point_states(921600);
// 获取当前坐标点的状态
auto& [last_class, last_save_time] = point_states[y * 1280 + x];
const auto now = std::chrono::steady_clock::now();
const auto duration_since_last_save =
std::chrono::duration_cast<std::chrono::seconds>(now - last_save_time).count();
auto need_save = false;
// 如果相同点且分类结果变化了，则保存
if (last_base_point == m_base_point && last_class != static_cast<int>(class_id)) {
bool need_save = false;
// 判断当前类别是否与上次保存的类别不同
if (last_class != class_id) {
Log.trace("Class changed", static_cast<int>(last_class), static_cast<int>(class_id));
need_save = true;
}
// 如果检测到新的基准点且结果为ready（y）或c类别，也保存
else if (last_base_point != m_base_point && (class_id == 2 || class_id == 0)) {
// y 或者 c
else if (class_id == 2 || class_id == 0) {
Log.trace("Class is", static_cast<int>(class_id));
need_save = true;
}
// 来点随机截图
else if (duration_since_last_save > 10) {
last_save_time = now;
// 长时间没变化，可能是被遮挡了
else if (duration_since_last_save > 30) {
Log.trace("Long time no change", duration_since_last_save);
need_save = true;
}
// 新增：如果最高得分低于阈值，则保存
if (score < 0.8f) {
Log.trace("Low score", score);
need_save = true;
}
if (need_save) {
std::string base_filename = utils::get_time_filestem() + "_" + std::to_string(m_base_point.x) + "_" +
std::to_string(m_base_point.y) + "(c" + std::to_string(prob[0]) + ")(n" +
std::to_string(prob[1]) + ")(y" + std::to_string(prob[2]) + ").png";
std::string subfolder;
switch (class_id) {
case 2:
subfolder = "y";
break;
case 1:
subfolder = "n";
break;
case 0:
subfolder = "c";
break;
default:
subfolder = "unknown";
break;
}
auto format_float = [](float val) {
return std::format("{:.3f}", val);
};
std::string subfolder = [class_id] {
switch (class_id) {
case 2:
return "y";
case 1:
return "n";
case 0:
return "c";
default:
return "unknown";
}
}();
std::string base_filename = utils::get_time_filestem() + "_" + std::to_string(m_base_point.x) + "_" +
std::to_string(m_base_point.y) + "(c" + format_float(prob[0]) + ")(n" +
format_float(prob[1]) + ")(y" + format_float(prob[2]) + ").png";
std::filesystem::path relative_path = utils::path("debug") / "skill_ready" / subfolder / base_filename;
last_class = class_id;
last_save_time = now;
std::filesystem::path relative_path =
utils::path("debug") / utils::path("skill_ready") / utils::path(subfolder) / base_filename;
last_base_point = m_base_point;
last_class = static_cast<int>(class_id);
Log.trace("Save image", relative_path);
asst::imwrite(relative_path, image);
}