-
根据指定窗口句柄截图
def WindowShot(self): """ 根据窗口句柄截图 返回: 图片对象 """ windll.user32.SetProcessDPIAware() hwnd = self.Handle left, top, right, bottom = win32gui.GetClientRect(hwnd) w = right - left h = bottom - top hwnd_dc = win32gui.GetWindowDC(hwnd) mfc_dc = win32ui.CreateDCFromHandle(hwnd_dc) save_dc = mfc_dc.CreateCompatibleDC() bitmap = win32ui.CreateBitmap() bitmap.CreateCompatibleBitmap(mfc_dc, w, h) save_dc.SelectObject(bitmap) # If Special K is running, this number is 3. If not, 1 result = windll.user32.PrintWindow(hwnd, save_dc.GetSafeHdc(), 3) bmpinfo = bitmap.GetInfo() bmpstr = bitmap.GetBitmapBits(True) img = np.frombuffer(bmpstr, dtype=np.uint8).reshape((bmpinfo["bmHeight"], bmpinfo["bmWidth"], 4)) img = np.ascontiguousarray(img)[..., :-1] # make image C_CONTIGUOUS and drop alpha channel #img = Image.frombuffer("RGB",(bmpinfo['bmWidth'], bmpinfo['bmHeight']),bmpstr, 'raw', 'BGRX', 0, 1) if not result: # result should be 1 win32gui.DeleteObject(bitmap.GetHandle()) save_dc.DeleteDC() mfc_dc.DeleteDC() win32gui.ReleaseDC(hwnd, hwnd_dc) raise RuntimeError(f"Unable to acquire screenshot! Result: {result}") #cv2.imwrite('./imgs/print.png', img) #return img return cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)
-
在图片中查找第一个相似的图片
def LocateOnImage(self, template, image=None, region=None, confidence=0.9, grayscale=True): """ 在image中寻找template，返回第一个查找到的范围 参数: template: 需要查找的图片，文件名或图片对象 image: 被查找的图片，文件名或图片对象 region: 查找范围 confidence: 置信度 grayscale: 是否为灰度图 返回值: 查找到的图片范围 """ if image is None: image = self.WindowShot() return pyautogui.locate(template, image, region=region, confidence=confidence, grayscale=grayscale)
-
在图片中查找所有相似的图片
def LocateAllOnImage(self, template, image=None, region=None, confidence=0.9, grayscale=True): """ 在image中寻找template，返回第一个查找到的范围 参数: template: 需要查找的图片，文件名或图片对象 image: 被查找的图片，文件名或图片对象 region: 查找范围 confidence: 置信度 grayscale: 是否为灰度图 返回值: 查找到的图片范围 """ if image is None: image = self.WindowShot() return pyautogui.locateAll(template, image, region=region, confidence=confidence, grayscale=grayscale)
-
识别手牌
def GetCards(self, image, player): hand_cards = [] cards = "" start_x = 0 width = player["width"] for card in self.AllCardsNC: confidence = player["confidence"] or 0.85 grayscale = True if(card in ('D', 'X')): confidence = 0.85 grayscale = False card_key = player["prefix"] + card matches = self.LocateAllOnImageName(card_key, image, player["region"], confidence, grayscale) if len(matches) > 0: sorted_matches = sorted(matches, key=lambda match: match[0]) #print(target_position) #print(mark) if(card != 'X'): start_x = 0 for match in sorted_matches: if(start_x == 0 or start_x + player["width"] < match[0]): #大小王容易判断错误，需要再判断一下颜色值 if(card == 'D'): #match = sorted_matches[0] y,x,h,w = match cropped_image = image[x:x+w, y:y+h] #colors = cv2.mean(cropped_image) mean, stddev = cv2.meanStdDev(cropped_image) #print("均值：", mean) #print("差值：", stddev) if(stddev[2][0] > 30): continue start_x = match[0] width = match[2] hand_cards.append({card:match}) cards += card return cards, hand_cards
-
将出牌记录发送给AI，进行预测
def GetCardsForPredict(self): #将出牌记录发送给AI，进行预测 resultStr = self.PostPredict() #获取最高胜率的出牌 last_move_cards = self.GetPredictWinRates(resultStr) return last_move_cards
-
出牌
def PlayedCards(self, cards, image=None): if image is not None: window_shot_image = image else: window_shot_image = self.WindowShot() player_self_cards, range = self.GetCards(window_shot_image, self.Players["PlayerSelf"]) select_index = [] select_cards = [] #将cards中的字符顺序反转 play_cards = cards[::-1] cards = play_cards for c in cards: index = -1 for card in player_self_cards: index += 1 if c == card and index not in select_index: select_index.append(index) select_cards.append(c) #print(range[index]) cards = cards.replace(c, '', 1) card_range = range[index][c] break cards = play_cards index = -1 #window_shot_image = self.WindowShot() for n in select_index: window_shot_image = self.WindowShot() player_self_cards, range = self.GetCards(window_shot_image, self.Players["PlayerSelf"]) if(len(player_self_cards) > n): #player_self_cards = player_self_cards index += 1 card_range = range[n][cards[index]] #如果要出的牌还未选择完毕，则点击 if(card_range[1] > self.CardTop): self.LeftClick(card_range) time.sleep(0.2) return self.GetSelectCards(play_cards)
-
DouZero 的请求结构体
Predict = { "bomb_num":0,#炸弹数量 "card_play_action_seq":'',#历史出牌动作序列，用逗号分隔 "last_move_landlord":'',#地主最后出的牌 "last_move_landlord_down":'',#地主下家最后出的牌 "last_move_landlord_up":'',#地主上家最后出的牌 "num_cards_left_landlord":20,#地主手牌剩余数量 "num_cards_left_landlord_down":17,#地主下家手牌剩余数量 "num_cards_left_landlord_up":17,#地主上家手牌剩余数量 "other_hand_cards":'',#还剩余的牌 "played_cards_landlord":'',#地主所有出的牌 "played_cards_landlord_down":'',#地主下家所有出的牌 "played_cards_landlord_up":'',#地主上家所有出的牌 "player_hand_cards":'',#玩家手中的牌 "player_position":0, #-当前玩家的位置序号 0 地主，1 地主下家，2 地主上家 "three_landlord_cards":''#三张底牌}
经验首页