public void ModelNotLoadedExceptionThrownTest()
{
var configuration = new ImageClassifierConfiguration
{
ImageHeight = 28, ImageWidth = 28,
Labels = new[] { "0", "1", "2", "3", "4", "5", "6", "7", "8", "9" },
PathToModel = "TestData/not_exist.pt"
};
var classifier = new MnistImageClassifier(configuration);
Assert.Throws <ModelNotLoadedException>(() => classifier.LoadModel());
}
public void RecognizeNumberTest()
{
var configuration = new ImageClassifierConfiguration
{
ImageHeight = 28, ImageWidth = 28,
Labels = new[] { "0", "1", "2", "3", "4", "5", "6", "7", "8", "9" },
PathToModel = "TestData/traced_model.pt"
};
var classifier = new MnistImageClassifier(configuration);
classifier.LoadModel();
var image = Image.FromFile("TestData/seven.png");
var label = classifier.ClassifyImage(image);
Assert.Contains(label, configuration.Labels);
}