Plumess
|
0896fcc0df
|
feat: 优化技能识别模型 (#11984)
* chore: 更新技能识别测试例 test_skill_ready()
1. 由于更新了技能识别为三分类模型，新增”可取消“类别，增加相关类别的测试；
2. 扩充了输出信息，便于Debug;
* feat: 调整技能识别推理函数，支持新的三分类模型
1. 新的三分类模型基于MobileNetv4训练并导出onnx，详情参考MaaAI仓库中的技能识别训练代码；
2. 新增”可取消“分类，标签为c，但暂不启用区分，与”未就绪“同样归于not ready;
3. 新增了部分前处理操作，以匹配模型的输入；
* perf: 使用 MobileNetv4 Small 重新训练了技能识别模型，改为三分类
新增”可取消“状态，即技能可能处于”可取消“，”未就绪“，”已就绪“三种状态；
新的三分类模型基于MobileNetv4 Conv Small 训练并导出，详情参考MaaAI仓库中的技能识别训练代码；
* feat: 调整技能识别自动截图，支持三分类
新增在debug模式下对”可取消“新类别的截图保存功能；
* chore: 存储截图时打印分数
* chore: 使用 ranges
* perf: 使用新训练集，基于官方mobilenetv4_conv_small的权重进行微调
16000张左右的训练集，在3000张测试集中准确率99.7%
---------
Co-authored-by: uye <99072975+ABA2396@users.noreply.github.com>
|
2025-04-29 14:30:24 +08:00 |
|
uye
|
ffa2334686
|
chore: 目录下存在 DEBUG_skill_ready.txt 时自动收集技能准备状态截图 (#11571)
* feat: 自动截图
* chore: 根目录下存在 DEBUG 或者 DEBUG.txt 时自动保存截图
* style: 优化格式
* chore: 目录下存在 DEBUG_skill_ready.txt 时自动保存截图
|
2025-01-19 01:28:56 +08:00 |
|
pre-commit-ci[bot]
|
ce02c80498
|
chore: Auto update by pre-commit hooks [skip changelog]
|
2024-10-22 08:40:28 +00:00 |
|
uye
|
da6b8e81fe
|
style: 格式化
[skip changelog]
|
2024-10-13 18:24:43 +08:00 |
|
Horror Proton
|
ec4242dbbb
|
chore: include path of onnxruntime>=1.16
|
2024-05-25 12:32:22 +08:00 |
|
MistEO
|
a8ec8709f4
|
refactor: 重构全部战斗相关视觉代码
|
2023-04-29 18:24:10 +08:00 |
|