解析数独游戏图片并给出答案

1. 加载并处理数独游戏图片

from PIL import Image
import pytesseract
import numpy as np
import cv2

# 加载数独游戏图片
image_path = "sudoku_image.jpg"
img = cv2.imread(image_path)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)[1]

2. 使用 pytesseract 进行数字识别

custom_config = r'--oem 3 --psm 10 outputbase digits'
text = pytesseract.image_to_string(thresh, config=custom_config)

print("Detected Sudoku Text:")
print(text)

3. 解数独游戏

def solve_sudoku(board):
    find = find_empty(board)
    if not find:
        return True
    else:
        row, col = find

    for num in range(1, 10):
        if is_valid(board, num, (row, col)):
            board[row][col] = num

            if solve_sudoku(board):
                return True

            board[row][col] = 0

    return False

def is_valid(board, num, pos):
    # 检查行
    for i in range(len(board[0])):
        if board[pos[0]][i] == num and pos[1] != i:
            return False

    # 检查列
    for i in range(len(board)):
        if board[i][pos[1]] == num and pos[0] != i:
            return False

    # 检查小方块
    box_x = pos[1] // 3
    box_y = pos[0] // 3

    for i in range(box_y * 3, box_y * 3 + 3):
        for j in range(box_x * 3, box_x * 3 + 3):
            if board[i][j] == num and (i, j) != pos:
                return False

    return True

def find_empty(board):
    for i in range(len(board)):
        for j in range(len(board[0])):
            if board[i][j] == 0:
                return (i, j)
    return None

# 初始化数独游戏数据结构并填入识别的数字
sudoku_board = np.zeros((9, 9), dtype=int)
idx = 0
for i in range(9):
    for j in range(9):
        if text[idx].isdigit():
            sudoku_board[i][j] = int(text[idx])
        idx += 1

# 解数独游戏
if solve_sudoku(sudoku_board):
    print("Sudoku Solution:")
    print(sudoku_board)
else:
    print("No solution found.")

---

这篇笔记概括了如何使用Python处理数独游戏图片、识别数字并解析数独游戏，最终给出数独游戏的解答。