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

Tyres FINALLY are starting to get recognized properly

Browse Source
This commit is contained in:
Rohmer Maxime
2023-04-08 09:21:24 +02:00
parent 08a2176385
commit ea3ca8cd28
2 changed files with 130 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
OCR_Decode/DriverTyresWindow.cs
+19 -16
View File
@@ -31,26 +31,16 @@ namespace OCR_Decode
private Tyre GetTyreInfos()
{
Bitmap tyreZone = GetSmallBitmapFromBigOne(WindowImage, FindTyreZone());
//MHIWsmhiw so we can detect Soft Medium Hard Inters and Wet tyres
//If it is the first lap with the tyre just the letter is displayed
string text = GetStringFromPng(tyreZone,"0123456789SMHIWsmhiw",OcrImage.WindowType.Tyre);
tyreZone.Save(Reader.DEBUG_DUMP_FOLDER + "ZONE" + ".png");
Tyre.Type type = Tyre.Type.Undefined;
int laps = -1;
//if (text.All(char.IsDigit))
//{
//We have a lap count
//if (text != "")
//{
//laps = Convert.ToInt32(text);
type = GetTyreTypeFromColor(OcrImage.GetAvgColorFromBitmap(OcrImage.Resize(tyreZone)));
//}
//}
/*
else
//MHIWsmhiw so we can detect Soft Medium Hard Inters and Wet tyres
string text = GetStringFromPng(tyreZone, "SMHIW", OcrImage.WindowType.Tyre);
if (text.Length == 1 && text != "")
{
//We have a tire type
//We found a tire letter
laps = 0;
text = text.ToUpper();
switch (text[0])
@@ -75,7 +65,20 @@ namespace OCR_Decode
break;
}
}
*/
else
{
string number = GetStringFromPng(tyreZone, "0123456789", OcrImage.WindowType.Tyre);
try
{
laps = Convert.ToInt32(number);
}
catch
{
laps = -1;
}
type = GetTyreTypeFromColor(OcrImage.GetAvgColorFromBitmap(OcrImage.Resize(tyreZone)));
}
tyreZone.Save(Reader.DEBUG_DUMP_FOLDER + "Tyre" + type + "Laps" + laps + '#' + rnd.Next(0, 1000) + ".png");
return new Tyre(type,laps);
}
OCR_Decode/OcrImage.cs
+108 -7
View File
@@ -48,14 +48,21 @@ namespace OCR_Decode
result = Dilatation(result, 1);
break;
case WindowType.Tyre:
result = RemoveBG(result);
result = Grayscale(result);
result = InvertColors(result);
result = RemoveUseless(result);
result = Resize(result);
result = Resize(result);
result = Resize(result);
result = Dilatation(result, 1);
//result = Grayscale(result);
//result = Tresholding(result,80);
result = Resize(result);
result = Dilatation(result, 1);
result = Resize(result);
result = Dilatation(result,1);
//result = InvertColors(result);
//result = Resize(result);
//result = Dilatation(result, 1);
//result = Resize(result);
//result = Dilatation(result,1);
//result = Erode(result,1);
//result = Resize(result);
@@ -169,6 +176,100 @@ namespace OCR_Decode
}
}
}
bmp.UnlockBits(bmpData);
return bmp;
}
public unsafe static Bitmap RemoveUseless(Bitmap bmp)
{
bmp = RemoveBG(bmp);
Rectangle rect = new Rectangle(0, 0, bmp.Width, bmp.Height);
BitmapData bmpData = bmp.LockBits(rect, ImageLockMode.ReadWrite, bmp.PixelFormat);
int bytesPerPixel = Bitmap.GetPixelFormatSize(bmp.PixelFormat) / 8;
byte* ptr = (byte*)bmpData.Scan0.ToPointer();
for (int y = 0; y < bmp.Height; y++)
{
byte* currentLine = ptr + (y * bmpData.Stride);
List<int> pixelsToRemove = new List<int>();
bool fromBorder = true;
for (int x = 0; x < bmp.Width; x++)
{
byte* pixel = currentLine + (x * bytesPerPixel);
int B = pixel[0];
int G = pixel[1];
int R = pixel[2];
if (fromBorder && B < 0x10 && G < 0x10 && R < 0x10)
{
pixelsToRemove.Add(x);
}
else
{
fromBorder = false;
}
}
fromBorder = true;
for (int x = bmp.Width - 1; x > 0; x--)
{
byte* pixel = currentLine + (x * bytesPerPixel);
int B = pixel[0];
int G = pixel[1];
int R = pixel[2];
if (fromBorder && B < 0x10 && G < 0x10 && R < 0x10)
{
pixelsToRemove.Add(x);
}
else
{
fromBorder = false;
}
}
foreach (int pxPos in pixelsToRemove)
{
byte* pixel = currentLine + (pxPos * bytesPerPixel);
pixel[0] = 0xFF;
pixel[1] = 0xFF;
pixel[2] = 0xFF;
}
}
//NOW REMOVING THE COLOR
Color limitColor = Color.FromArgb(0x50, 0x50, 0x50);
for (int y = 0; y < bmp.Height; y++)
{
byte* currentLine = ptr + (y * bmpData.Stride);
for (int x = 0; x < bmp.Width; x++)
{
byte* pixel = currentLine + (x * bytesPerPixel);
int B = pixel[0];
int G = pixel[1];
int R = pixel[2];
//We remove the background pixels
if (R >= limitColor.R || G >= limitColor.G || B >= limitColor.B)
{
pixel[0] = 0xFF;
pixel[1] = 0xFF;
pixel[2] = 0xFF;
}
}
}
bmp.UnlockBits(bmpData);
return bmp;