private hogwartsHouse ConvertIdx(int idx)
{
hogwartsHouse house = hogwartsHouse.Gryffindor;
if (idx == 0)
{
house = hogwartsHouse.Gryffindor;
}
else if (idx == 1)
{
house = hogwartsHouse.Hufflpuff;
}
else if (idx == 2)
{
house = hogwartsHouse.Ravenclaw;
}
else if (idx == 3)
{
house = hogwartsHouse.Slytherin;
}
return(house);
}
public bool Sort(Mat inputData, ref hogwartsHouse house, ref float value)
{
Mat blob = Dnn.blobFromImage(inputData);
if (blob == null)
{
return(false);
}
net.setInput(blob);
Mat prob = net.forward();
if (prob == null)
{
return(false);
}
int idx = 0;
(idx, value) = get_max_idx(prob);
house = ConvertIdx(idx);
return(true);
}
