mirror of
https://github.com/babalae/better-genshin-impact.git
synced 2026-05-27 10:15:50 +08:00
0c02808626
* 使用TorchSharp重写RodNet，以利后续优化 * 增加一个外部torch加载配置来代替直接的依赖，如配置不生效则使用原先手搓的算法 * BgiOnnxFactory取消单例，改为在App服务类中注册为单例，由此修复了一堆单元测试 * BgiOnnxFactory中几个静态方法改为成员方法以和App解耦；因不再有多个mat源供消耗，FishBite中文字块算法不再改动传入的mat，使得后续串联的算法不受其影响 * 将BehavioursTests中临时的配置读取方式改为读取主项目编译环境中的json文件；新建单元测试的README * 将RodNet算法更新到010006a44c的版本；RodNet中关于torch库推理和直接数学计算的校验移至单元测试 * 更新RodNet算法至最新：add5672731* 注释调试用的代码
82 lines
3.2 KiB
C#
82 lines
3.2 KiB
C#
using OpenCvSharp;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using BetterGenshinImpact.Core.Recognition.OCR;
|
|
using System.Drawing;
|
|
using OpenCvSharp.Extensions;
|
|
|
|
namespace BetterGenshinImpact.UnitTest.CoreTests.RecognitionTests.OCRTests
|
|
{
|
|
[Collection("Init Collection")]
|
|
public partial class PaddleOcrServiceTests
|
|
{
|
|
private readonly PaddleFixture paddle;
|
|
public PaddleOcrServiceTests(PaddleFixture paddle)
|
|
{
|
|
this.paddle = paddle;
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData("zh-Hans", "挑战,达成", "挑战.*达成")]
|
|
[InlineData("zh-Hans", "凯瑟琳")]
|
|
[InlineData("en", "Daily")]
|
|
[InlineData("en", "Katheryne")]
|
|
[InlineData("en", "Clickany where to close", "Click.*any.*where.*to.*close")]
|
|
[InlineData("en", "2-Star Artifacts", "2-Star.*Artifacts")]
|
|
[InlineData("zh-Hant", "凱瑟琳")]
|
|
[InlineData("zh-Hant", "委託", "委[託話]")]
|
|
[InlineData("zh-Hant", "挑戰,達成", "挑戰.*達成")]
|
|
[InlineData("zh-Hant", "自動,退出", "自動.*退出")]
|
|
[InlineData("zh-Hant", "跳過")]
|
|
[InlineData("zh-Hant", "地脈異常", "地[脈服][異翼昊]常")]
|
|
[InlineData("zh-Hant", "點擊任意位置關閉")]
|
|
[InlineData("zh-Hant", "快速選擇", "快速選擇")]
|
|
[InlineData("zh-Hant", "二星聖遺物")]
|
|
[InlineData("fr", "quotidien")]
|
|
[InlineData("fr", "Expédition")]
|
|
[InlineData("fr", "Pêcher", "P[êé]cher")]
|
|
[InlineData("fr", "Synthèse", "Synth[èé](se|tiser)")]
|
|
[InlineData("fr", "Synthétiser", "Synth[èé](se|tiser)")]
|
|
[InlineData("fr", "Défi, terminé", "Défi.*terminé")]
|
|
[InlineData("fr", "Sortie")]
|
|
[InlineData("fr", "Passer")]
|
|
[InlineData("fr", "Anomalie énergétique", "Anomalie.*énergétique")]
|
|
[InlineData("fr", "Cliquez pour fermer", "Cliquez.*pour.*fermer")]
|
|
[InlineData("fr", "Sélection rapide", "Sélection.*rapide")]
|
|
[InlineData("fr", "Artéfact 2★", "Artéfact.*2★?")]
|
|
/// <summary>
|
|
/// 测试识别各种文字，结果为成功
|
|
/// </summary>
|
|
public void PaddleOcrService_VariousLangWords_ShouldEqualOrMatch(string cultureInfoName, string word, string? pattern = null)
|
|
{
|
|
//
|
|
using Bitmap bitmap = new Bitmap(word.Length * 50, 100);
|
|
|
|
using Graphics g = Graphics.FromImage(bitmap);
|
|
g.Clear(Color.White);
|
|
using Font font = new Font("Arial", 12);
|
|
using Brush brush = new SolidBrush(Color.Black);
|
|
g.DrawString(word, font, brush, 0, 50);
|
|
using Mat mat4 = bitmap.ToMat();
|
|
using Mat mat = mat4.CvtColor(ColorConversionCodes.RGBA2RGB);
|
|
|
|
//
|
|
PaddleOcrService sut = paddle.Get(cultureInfoName);
|
|
string actual = sut.Ocr(mat).Replace(" ", "");
|
|
|
|
//
|
|
if (pattern == null)
|
|
{
|
|
Assert.Equal(word, actual);
|
|
}
|
|
else
|
|
{
|
|
Assert.Matches(pattern, actual);
|
|
}
|
|
}
|
|
}
|
|
}
|