代码收藏家技术教程 2025-02-25

斗地主老是输？一起用 Python 做个 AI 出牌器！

磐创AI 2022年04月15日 23:34

前言

最近在网上看到一个有意思的开源项目，基于快手团队开发的开源AI斗地主——DouZero做的一个“成熟”的AI。

今天我们就一起来学习下是如何制作一个基于DouZero的出牌器，看看AI是如何来帮助斗地主的！

核心功能设计

首先这款出牌器是基于DouZero开发的，核心是需要利用训练好的AI模型来帮住我们，给出最优出牌方案。

其次关于出牌器，先要需要确认一个AI出牌角色，代表我们玩家自己。我们只要给这个AI输入玩家手牌和三张底牌。确认好地主和农民的各个角色，告诉它三个人对应的关系，这样就可以确定队友和对手。

我们还要将每一轮其他两人的出牌输入，这样出牌器就可以根据出牌数据，及时提供给我们最优出牌决策，带领我们取得胜利！

那么如何获取三者之间的关系呢？谁是地主？谁是农民？是自己一人作战还是农民合作？自己玩家的手牌是什么？三张底牌是什么？这些也都需要在开局后确认好。

大致可以整理出要实现的核心功能如下：

UI设计排版布局

显示三张底牌

显示AI角色出牌数据区域，上家出牌数据区域，下家出牌数据区域，本局胜率区域

AI玩家手牌区域

AI出牌器开始停止

手牌和出牌数据识别

游戏刚开始根据屏幕位置，截图识别AI玩家手牌及三张底牌

确认三者之间的关系，识别地主和农民角色，确认队友及对手关系

识别每轮三位玩家依次出了什么牌，刷新显示对应区域

AI出牌方案输出

加载训练好的AI模型，初始化游戏环境

每轮出牌判断，根据上家出牌数据给出最优出牌决策

自动刷新玩家剩余手牌和本局胜率预测

实现步骤

1. UI设计排版布局

根据上述功能，首先考虑进行简单的UI布局设计，使用的是pyqt5。核心设计代码如下：

def setupUi(self, Form):    Form.setObjectName("Form")    Form.resize(440, 395)    font = QtGui.QFont()    font.setFamily("Arial")    font.setPointSize(9)    font.setBold(True)    font.setItalic(False)    font.setWeight(75)    Form.setFont(font)    self.WinRate = QtWidgets.QLabel(Form)    self.WinRate.setGeometry(QtCore.QRect(240, 180, 171, 61))    font = QtGui.QFont()    font.setPointSize(14)    self.WinRate.setFont(font)    self.WinRate.setAlignment(QtCore.Qt.AlignCenter)    self.WinRate.setObjectName("WinRate")    self.InitCard = QtWidgets.QPushButton(Form)    self.InitCard.setGeometry(QtCore.QRect(60, 330, 121, 41))    font = QtGui.QFont()    font.setFamily("Arial")    font.setPointSize(14)    font.setBold(True)    font.setWeight(75)    self.InitCard.setFont(font)    self.InitCard.setStyleSheet("")    self.InitCard.setObjectName("InitCard")    self.UserHandCards = QtWidgets.QLabel(Form)    self.UserHandCards.setGeometry(QtCore.QRect(10, 260, 421, 41))    font = QtGui.QFont()    font.setPointSize(14)    self.UserHandCards.setFont(font)    self.UserHandCards.setAlignment(QtCore.Qt.AlignCenter)    self.UserHandCards.setObjectName("UserHandCards")    self.LPlayer = QtWidgets.QFrame(Form)    self.LPlayer.setGeometry(QtCore.QRect(10, 80, 201, 61))    self.LPlayer.setFrameShape(QtWidgets.QFrame.StyledPanel)    self.LPlayer.setFrameShadow(QtWidgets.QFrame.Raised)    self.LPlayer.setObjectName("LPlayer")    self.LPlayedCard = QtWidgets.QLabel(self.LPlayer)    self.LPlayedCard.setGeometry(QtCore.QRect(0, 0, 201, 61))    font = QtGui.QFont()    font.setPointSize(14)    self.LPlayedCard.setFont(font)    self.LPlayedCard.setAlignment(QtCore.Qt.AlignCenter)    self.LPlayedCard.setObjectName("LPlayedCard")    self.RPlayer = QtWidgets.QFrame(Form)    self.RPlayer.setGeometry(QtCore.QRect(230, 80, 201, 61))    font = QtGui.QFont()    font.setPointSize(16)    self.RPlayer.setFont(font)    self.RPlayer.setFrameShape(QtWidgets.QFrame.StyledPanel)    self.RPlayer.setFrameShadow(QtWidgets.QFrame.Raised)    self.RPlayer.setObjectName("RPlayer")    self.RPlayedCard = QtWidgets.QLabel(self.RPlayer)    self.RPlayedCard.setGeometry(QtCore.QRect(0, 0, 201, 61))    font = QtGui.QFont()    font.setPointSize(14)    self.RPlayedCard.setFont(font)    self.RPlayedCa