ML.NET - Image classification program always returns a wrong prediction

Viewed 35

I am trying to create an image classification program using the ML.NET library. I am following the steps described in the Microsoft tutorial below.

I loaded 100 red images and 100 blue images. The training phase seems to be working great as it was able to make predictions with 100% accuracy. The problem happens when using the model; it always returns the wrong prediction and the score is always 0. What could cause that issue?

ML.NET

1 Answers

Solved my problem. The issue was that my input data had the image path but a byte array of the image was also needed.

Related