当前位置：首页 > news >正文

用Python+OpenCV给斗地主做个‘外挂’：手把手教你写个桌面记牌器（附源码）

news 2026/5/12 20:49:53

Python+OpenCV实战：打造智能斗地主记牌工具

记得去年宿舍通宵打牌时，室友总抱怨记不住出过的牌。作为计算机系学生，我琢磨着能否用课堂学的OpenCV做个记牌工具。三周后，当这个能自动识别桌面上扑克牌的小程序成功运行时，那种将理论知识转化为实际应用的成就感，至今难忘。下面就把这个结合图像处理与界面开发的完整项目经验分享给大家。

1. 开发环境与工具链搭建

工欲善其事，必先利其器。这个项目需要多个Python库的协同工作，建议使用Anaconda创建独立环境：

conda create -n poker_ai python=3.8 conda activate poker_ai pip install opencv-python numpy pyqt5 pillow pywin32

核心工具链构成：

工具	版本	用途
OpenCV	4.5+	图像采集与处理核心库
PyQt5	5.15+	构建用户界面
NumPy	1.20+	数值计算支持
Pillow	8.0+	图像格式转换
pywin32	300+	Windows系统API调用

提示：避免混用pip和conda安装，可能导致依赖冲突。建议全部使用pip在虚拟环境中安装。

常见环境问题解决方案：

出现DLL加载错误：安装VC++ 2015-2019运行库
摄像头无法调用：检查权限并更新驱动程序
PyQt5样式异常：安装Qt官方运行时库

2. 屏幕图像采集与预处理

精准获取游戏窗口是第一步。我们采用Windows API获取指定窗口的像素数据：

import win32gui import win32ui from ctypes import windll def capture_window(hwnd): left, top, right, bottom = win32gui.GetWindowRect(hwnd) w = right - left h = bottom - top hwndDC = win32gui.GetWindowDC(hwnd) mfcDC = win32ui.CreateDCFromHandle(hwndDC) saveDC = mfcDC.CreateCompatibleDC() saveBitMap = win32ui.CreateBitmap() saveBitMap.CreateCompatibleBitmap(mfcDC, w, h) saveDC.SelectObject(saveBitMap) windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 2) bmpinfo = saveBitMap.GetInfo() bmpstr = saveBitMap.GetBitmapBits(True) img = np.frombuffer(bmpstr, dtype='uint8').reshape((h, w, 4)) # 释放资源 win32gui.DeleteObject(saveBitMap.GetHandle()) saveDC.DeleteDC() mfcDC.DeleteDC() win32gui.ReleaseDC(hwnd, hwndDC) return cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)

关键预处理步骤：

色彩空间转换：BGR→HSV便于颜色阈值处理
高斯模糊：3×3内核消除噪声
边缘检测：Canny算法提取牌面轮廓
形态学操作：闭运算填充内部空隙

def preprocess_image(img): hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) blur = cv2.GaussianBlur(hsv, (3,3), 0) edges = cv2.Canny(blur, 50, 150) kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5,5)) closed = cv2.morphologyEx(edges, cv2.MORPH_CLOSE, kernel) return closed

3. 扑克牌识别核心技术

3.1 牌面区域定位

通过轮廓分析找到所有可能的扑克牌区域：

def find_card_contours(processed_img): contours, _ = cv2.findContours( processed_img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE ) cards = [] for cnt in contours: peri = cv2.arcLength(cnt, True) approx = cv2.approxPolyDP(cnt, 0.02*peri, True) if len(approx) == 4 and cv2.contourArea(cnt) > 1000: x,y,w,h = cv2.boundingRect(approx) aspect_ratio = w / float(h) if 0.6 <= aspect_ratio <= 0.8: cards.append((x,y,w,h)) return sorted(cards, key=lambda c: c[0]) # 按x坐标排序

3.2 牌面字符识别

采用模板匹配与OCR结合的方式：

花色识别（模板匹配）：

def detect_suit(card_img): suits = ['hearts', 'diamonds', 'clubs', 'spades'] best_match = None max_val = 0 for suit in suits: template = cv2.imread(f'templates/{suit}.png', 0) res = cv2.matchTemplate(card_gray, template, cv2.TM_CCOEFF_NORMED) _, val, _, _ = cv2.minMaxLoc(res) if val > max_val: max_val = val best_match = suit return best_match if max_val > 0.7 else None

数字/字母识别（OCR预处理流程）：
- 提取ROI区域
- 二值化处理
- 字符分割
- 使用预训练CNN模型识别

4. PyQt5界面设计与功能集成

创建带记忆功能的用户界面：

from PyQt5.QtWidgets import (QApplication, QMainWindow, QLabel, QPushButton, QVBoxLayout, QWidget) class PokerTracker(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle("斗地主智能记牌器") self.setGeometry(100, 100, 400, 600) self.central_widget = QWidget() self.layout = QVBoxLayout() self.status_label = QLabel("等待游戏开始...") self.start_btn = QPushButton("开始监控") self.start_btn.clicked.connect(self.start_tracking) self.layout.addWidget(self.status_label) self.layout.addWidget(self.start_btn) self.central_widget.setLayout(self.layout) self.setCentralWidget(self.central_widget) self.played_cards = set() self.current_hand = [] def start_tracking(self): self.status_label.setText("监控中...") # 启动定时截图线程

界面功能亮点：

实时牌面显示：可视化已出牌和剩余牌
概率计算：根据已出牌计算各玩家持牌概率
历史记录：保存多局比赛数据
热键支持：Alt+S快速截图

5. 实战中的坑与解决方案

5.1 特效干扰问题

当玩家出顺子或炸弹时，游戏会显示特效动画，导致截图失真。解决方案：

检测动画帧（通过颜色直方图突变）
延迟100ms后重新截图
添加异常重试机制

5.2 多分辨率适配

不同显示器分辨率导致牌面位置变化。我们采用：

动态检测窗口尺寸
比例计算关键坐标
自适应模板缩放

def adapt_resolution(window_size): base_w, base_h = 1920, 1080 # 基准分辨率 curr_w, curr_h = window_size scale_x = curr_w / base_w scale_y = curr_h / base_h return { 'card_width': int(80 * scale_x), 'card_height': int(120 * scale_y), 'margin_x': int(30 * scale_x), 'margin_y': int(50 * scale_y) }