mirror of
https://github.com/babalae/better-genshin-impact.git
synced 2026-05-25 10:05:49 +08:00
37 lines
1.4 KiB
C#
37 lines
1.4 KiB
C#
using BetterGenshinImpact.Core.Config;
|
|
using BetterGenshinImpact.Core.Recognition.OCR;
|
|
using BetterGenshinImpact.Core.Recognition.ONNX;
|
|
using BetterGenshinImpact.UnitTest.CoreTests.RecognitionTests.OCRTests;
|
|
|
|
namespace BetterGenshinImpact.UnitTest.GameTaskTests.AutoFishingTests
|
|
{
|
|
[Collection("Init Collection")]
|
|
public partial class BehavioursTests
|
|
{
|
|
#pragma warning disable CS8618 // 在退出构造函数时,不可为 null 的字段必须包含非 null 值。请考虑添加 "required" 修饰符或声明为可为 null。
|
|
private static BgiYoloPredictor predictor;
|
|
#pragma warning restore CS8618 // 在退出构造函数时,不可为 null 的字段必须包含非 null 值。请考虑添加 "required" 修饰符或声明为可为 null。
|
|
|
|
private readonly PaddleFixture paddle;
|
|
|
|
public BehavioursTests(PaddleFixture paddle, TorchFixture torch)
|
|
{
|
|
this.paddle = paddle;
|
|
this.useTorch = torch.UseTorch;
|
|
}
|
|
|
|
private IOcrService OcrService => paddle.Get();
|
|
|
|
private static BgiYoloPredictor Predictor
|
|
{
|
|
get
|
|
{
|
|
return LazyInitializer.EnsureInitialized(ref predictor,
|
|
() => new BgiOnnxFactory(new FakeLogger<BgiOnnxFactory>())
|
|
.CreateYoloPredictor(BgiOnnxModel.BgiFish));
|
|
}
|
|
}
|
|
|
|
private readonly bool useTorch;
|
|
}
|
|
} |