fix(single-mode): error when recognize +100

fix #162
NateScarlet · Aug 23, 2021 · 15509d6 · 15509d6
1 parent 4a9d72c
commit 15509d6
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/...training/__snapshots__/training_test/test_from_training_scene.training_scene_issue162.txt b/...training/__snapshots__/training_test/test_from_training_scene.training_scene_issue162.txt
@@ -0,0 +1 @@
+Training<lv=5 spd=100 pow=57 ski=19 ptn=spd@5,wis@5,spd@5,spd@4,spd@4>
diff --git a/auto_derby/single_mode/training/training.py b/auto_derby/single_mode/training/training.py
@@ -113,6 +113,16 @@ def _ocr_training_effect(img: Image) -> int:
         cv2.waitKey()
         cv2.destroyAllWindows()
 
+    # +100 has different color
+    hash100 = "0000000000006066607770ff70df60df00000000000000000000000000000000"
+    if (
+        imagetools.compare_hash(
+            imagetools.image_hash(imagetools.pil_image(text_img)),
+            hash100,
+        )
+        > 0.9
+    ):
+        return 100
     text = ocr.text(image_from_array(text_img))
     if not text:
         return 0

diff --git a/auto_derby/test_data/single_mode/training_scene_issue162.png b/auto_derby/test_data/single_mode/training_scene_issue162.png