fix(single-mode): error when recognize race

NateScarlet · Jun 15, 2021 · 56032df · 56032df
1 parent 8dbcb33
commit 56032df
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 3 deletions.
diff --git a/auto_derby/ocr.py b/auto_derby/ocr.py
@@ -193,7 +193,7 @@ def _get_expanded_bbox(index: int) -> Tuple[int, int, int, int]:
             char_parts
             and l > char_non_zero_bbox[2]
             and (
-                l - char_non_zero_bbox[0] > max_char_width * 0.6
+                l - char_non_zero_bbox[0] > max_char_width * 0.7
                 or l - char_non_zero_bbox[2] > max_char_width * 0.2
                 or r - char_non_zero_bbox[0] > max_char_width
                 or (

diff --git a/auto_derby/single_mode/race_test.py b/auto_derby/single_mode/race_test.py
@@ -129,6 +129,16 @@ def test_find_by_race_detail_image_9():
     assert race1.stadium == "京都", race1.stadium
 
 
+def test_find_by_race_detail_image_10():
+    img = PIL.Image.open(_test.DATA_PATH / "race_detail_10.png").convert("RGB")
+    ctx = Context()
+    ctx.date = (1, 9, 1)
+    race1 = race.find_by_race_detail_image(ctx, img)
+
+    assert race1.name == "小倉ジュニアステークス", race1.name
+    assert race1.stadium == "小倉", race1.stadium
+
+
 def test_find_by_race_detail_image_issue31():
     img = (
         PIL.Image.open(_test.DATA_PATH / "race_detail_issue31.png")

diff --git a/auto_derby/single_mode/test_data/race_detail_10.png b/auto_derby/single_mode/test_data/race_detail_10.png
diff --git a/ocr_labels.json b/ocr_labels.json
@@ -643,5 +643,7 @@
   "00007e0078cc30fc18feffcff7fff8ff18fe18fc7efc8ffc8fcdfffdff7c0008": "離",
   "fe3ffe7fe001e001e001feff86e706c386c3f6ff9eff86e186e1fefffefffc7f": "函",
   "00003818fe0fffc7fec77efe66fe76fe7e7e7e067e063e3e22fe7efefeff427e": "館",
-  "f80ffc1f3e7c1e780f700f701ff83efcfcffe0fb0070007808783e7efc3ff80f": "9"
+  "f80ffc1f3e7c1e780f700f701ff83efcfcffe0fb0070007808783e7efc3ff80f": "9",
+  "800380038003983b9c3b9c7b9c7b9e739e738ff387e387438003e003f003f001": "小",
+  "8001e007f83ffefffefffc7ffc3f1c301c301c38fc3ffe3ffefffefff8ffe07f": "倉"
 }
diff --git a/single_mode_choices.json b/single_mode_choices.json
@@ -50,5 +50,7 @@
   "03570efc489d51383e403812360ffb8b": 2,
   "c3dec6e3c4b4f3158b1c555bee9755e6": 1,
   "da09be83fc01e842ac2dcd6790849c0d": 2,
-  "b846741f4d033493165584ef05567b5a": 1
+  "b846741f4d033493165584ef05567b5a": 1,
+  "ca0db1e260ab5c8abf6dc712f3d88db3": 2,
+  "8fde2da6a43d9b2d5156892baa50e498": 2
 }