diff --git a/board_detector.py b/board_detector.py index 6c5cf943ead50ad862cda03cd441a05e1e673671..65d52270e7f1f11b1f4a35431b226c3b600111fe 100644 --- a/board_detector.py +++ b/board_detector.py @@ -57,8 +57,8 @@ def find_longest_lines(img): filtered_horizontal = filter_lines(horizontal_line_points, 50) # get the 9 largest lines - sorted_vertical = sorted(filtered_vertical, key=lambda line: min(line[0][1], line[0][3]))[:12] - sorted_horizontal = sorted(filtered_horizontal, key=lambda line: min(line[0][0], line[0][2]))[:12] + sorted_vertical = sorted(filtered_vertical, key=lambda line: min(line[0][1], line[0][3]))[:9] + sorted_horizontal = sorted(filtered_horizontal, key=lambda line: min(line[0][0], line[0][2]))[:9] return sorted_vertical, sorted_horizontal diff --git a/game.py b/game.py index 7ede794f61688174cea08d0e2147b92a5a20556d..3364da0f8c013f5ee80d21bdbc305875ee8a520c 100644 --- a/game.py +++ b/game.py @@ -76,7 +76,7 @@ class ChessGame: img = cv2.resize(cropped_img, (512, 512)) # img = orig_img - if (self.show_cv): + if (self.show_cv and not self.calibrate): display_img([orig_img, img]) # display_img([img]) diff --git a/ocr_test_images/ocr_kqbp_flipped1.jpg b/ocr_test_images/ocr_kqbp_flipped1.jpg new file mode 100644 index 0000000000000000000000000000000000000000..6915d547c6a0c5fc6919d4f5310e0dc96de1b4cd Binary files /dev/null and b/ocr_test_images/ocr_kqbp_flipped1.jpg differ diff --git a/ocr_test_images/ocr_kqbp_flipped2.jpg b/ocr_test_images/ocr_kqbp_flipped2.jpg new file mode 100644 index 0000000000000000000000000000000000000000..733db1b9457d2bfa97a66c3c25712b138a1e0390 Binary files /dev/null and b/ocr_test_images/ocr_kqbp_flipped2.jpg differ diff --git a/ocr_test_images/ocr_kqbp_flipped3.jpg b/ocr_test_images/ocr_kqbp_flipped3.jpg new file mode 100644 index 0000000000000000000000000000000000000000..84949c7e3f2eb6f5b1fb7352c29f65bbc263d4bc Binary files /dev/null and b/ocr_test_images/ocr_kqbp_flipped3.jpg differ diff --git a/ocr_test_images/ocr_kqbp_flipped4.jpg b/ocr_test_images/ocr_kqbp_flipped4.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c51a8045248c74976af284263e4c20513108b36d Binary files /dev/null and b/ocr_test_images/ocr_kqbp_flipped4.jpg differ diff --git a/ocr_test_images/ocr_kqbp_flipped5.jpg b/ocr_test_images/ocr_kqbp_flipped5.jpg new file mode 100644 index 0000000000000000000000000000000000000000..4ab6a55ac59610f3ac4a75d1fb4c80ab00362400 Binary files /dev/null and b/ocr_test_images/ocr_kqbp_flipped5.jpg differ