Maxluli/OCR_Decode
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

Now the color recognition is better

Browse Source
This commit is contained in:
Rohmer Maxime
2023-04-08 08:45:57 +02:00
parent c515e5ff62
commit 08a2176385
2 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
OCR_Decode/DriverTyresWindow.cs
+10 -8
View File
@@ -38,15 +38,16 @@ namespace OCR_Decode
tyreZone.Save(Reader.DEBUG_DUMP_FOLDER + "ZONE" + ".png");
Tyre.Type type = Tyre.Type.Undefined;
int laps = -1;
if (text.All(char.IsDigit))
{
//if (text.All(char.IsDigit))
//{
//We have a lap count
if (text != "")
{
laps = Convert.ToInt32(text);
type = GetTyreTypeFromColor(OcrImage.GetAvgColorFromBitmap(tyreZone));
}
}
//if (text != "")
//{
//laps = Convert.ToInt32(text);
type = GetTyreTypeFromColor(OcrImage.GetAvgColorFromBitmap(OcrImage.Resize(tyreZone)));
//}
//}
/*
else
{
//We have a tire type
@@ -74,6 +75,7 @@ namespace OCR_Decode
break;
}
}
*/
tyreZone.Save(Reader.DEBUG_DUMP_FOLDER + "Tyre" + type + "Laps" + laps + '#' + rnd.Next(0, 1000) + ".png");
return new Tyre(type,laps);
}
OCR_Decode/OcrImage.cs
+3 -2
View File
@@ -201,7 +201,8 @@ namespace OCR_Decode
int R = pixel[2];
//We remove the background pixels
if (R >= limitColor.R && G >= limitColor.G && B >= limitColor.B)
if (R >= limitColor.R || G >= limitColor.G || B >= limitColor.B)
{
totPixels++;
totB += pixel[0];
@@ -213,7 +214,7 @@ namespace OCR_Decode
}
bmp.UnlockBits(bmpData);
return Color.FromArgb(255,totR / totPixels,totG / totPixels,totB / totPixels);
return Color.FromArgb(255,Convert.ToInt32((float)totR / (float)totPixels),Convert.ToInt32((float)totG / (float)totPixels),Convert.ToInt32((float)totB / (float)totPixels));
}
public static Bitmap InvertColors(Bitmap input)
{