python实现连连看辅助之图像识别延伸


Posted in Python onJuly 17, 2019

python实现连连看辅助?图像识别延伸(百度AI),供大家参考,具体内容如下

百度AI平台提供图片相似检索API接口,并有详细的API文档说明,可以更好的实现图片识别。

from aip import AipImageSearch

""" 你的 APPID AK SK """
APP_ID = '***'
API_KEY = '***'
SECRET_KEY = '***'

client = AipImageSearch(APP_ID, API_KEY, SECRET_KEY)
with open("{}-{}.jpg".format(1, 1), "rb") as f:
 im = f.read()
# im = self.image_list[row][col]
# 将图片与百度云自建相似图库中的图片对比相似度
res = client.similarSearch(im)
for r in res["result"]:
 if r["score"] > 0.9:
  print(r["brief"])

百度AI平台提供非常多的API接口,值得研究。

代码

import win32gui
import time
from PIL import ImageGrab , Image
import numpy as np
from pymouse import PyMouse
from aip import AipImageSearch


class GameAuxiliaries(object):
 def __init__(self):
  self.wdname = r'宠物连连看经典版2,宠物连连看经典版2小游戏,4399小游戏 www.4399.com - Google Chrome'
  # self.wdname = r'main.swf - PotPlayer'
  self.image_list = {}
  self.m = PyMouse()
  self.APP_ID = '15633871'
  self.API_KEY = 'LNMuXHmULcZM0PRKX8ZT4OnB'
  self.SECRET_KEY = 'IwvyYxeDLIR5XvEmnX3ENWoVzMITkdBL'

  self.client = AipImageSearch(self.APP_ID, self.API_KEY, self.SECRET_KEY)


 def find_game_wd(self,wdname):
  # 取得窗口句柄
  hdwd = win32gui.FindWindow(0,wdname)
  # 设置为最前显示
  win32gui.SetForegroundWindow(hdwd)
  time.sleep(1)

 def get_img(self):
  image = ImageGrab.grab((417, 289, 884, 600))
  # image = ImageGrab.grab((417, 257, 885, 569))
  image.save('1.jpg','JPEG')
  for x in range(1,9):
   self.image_list[x] = {}
   for y in range(1,13):
    top = (x - 1) * 38 + (x-2)
    left =(y - 1) * 38 +(y-2)
    right = y * 38 + (y-1)
    bottom = x * 38 +(x -1)
    if top < 0:
     top = 0
    if left < 0 :
     left = 0
    im_temp = image.crop((left,top,right,bottom))
    im = im_temp.crop((1,1,37,37))
    im.save('{}-{}.jpg'.format(x,y))
    self.image_list[x][y]=im

 def compare_img_baiduapi(self,im):
  '''将图片与百度云自建相似图库中的图片对比相似度'''
  pass

 # 判断两个图片是否相同。汉明距离,平均哈希
 def compare_img(self,im1,im2):
  img1 = im1.resize((20, 20), Image.ANTIALIAS).convert('L')
  img2 = im2.resize((20, 20), Image.ANTIALIAS).convert('L')
  pi1 = list(img1.getdata())
  pi2 = list(img2.getdata())
  avg1 = sum(pi1) / len(pi1)
  avg2 = sum(pi2) / len(pi2)
  hash1 = "".join(map(lambda p: "1" if p > avg1 else "0", pi1))
  hash2 = "".join(map(lambda p: "1" if p > avg2 else "0", pi2))
  match = 0
  for i in range(len(hash1)):
   if hash1[i] != hash2[i]:
    match += 1
  # match = sum(map(operator.ne, hash1, hash2))
  # match 值越小,相似度越高
  return match


 # 将图片矩阵转换成数字矩阵

 def create_array(self):
  array = np.zeros((10,14),dtype=np.int32)
  img_type_list = []
  for row in range(1,len(self.image_list)+1):
   for col in range(1,len(self.image_list[1])+1):
    # im = Image.open('{}-{}.jpg'.format(row,col))
    with open("{}-{}.jpg".format(row,col), "rb") as f:
     im = f.read()
    # im = self.image_list[row][col]
    # 将图片与百度云自建相似图库中的图片对比相似度
    res = self.client.similarSearch(im)
    while len(res) == 2:

     res = self.client.similarSearch(im)
     print(res)
     print(row, col)
     time.sleep(0.2)
    print(row,col)
    for r in res["result"]:
     if r["score"] > 0.9:
      array[row][col]=r["brief"]

  return array

 def row_zero(self,x1,y1,x2,y2,array):
  '''相同的图片中间图标全为空'''
  if x1 == x2:
   min_y = min(y1,y2)
   max_y = max(y1,y2)
   if max_y - min_y == 1:
    return True
   for y in range(min_y+1,max_y):
    if array[x1][y] != 0 :
     return False
   return True
  else:
   return False

 def col_zero(self,x1,y1,x2,y2,array):
  '''相同的图片同列'''
  if y1 == y2:
   min_x = min(x1,x2)
   max_x = max(x1,x2)
   if max_x - min_x == 1:
    return True
   for x in range(min_x+1,max_x):
    if array[x][y1] != 0 :
     return False
   return True
  else:
   return False

 def two_line(self,x1,y1,x2,y2,array):
  '''两条线相连,转弯一次'''
  for row in range(1,9):
   for col in range(1,13):
    if row == x1 and col == y2 and array[row][col]==0 and self.row_zero(x1,y1,row,col,array) and self.col_zero(x2,y2,row,col,array):
     return True
    if row == x2 and col == y1 and array[row][col]==0 and self.row_zero(x2,y2,row,col,array) and self.col_zero(x1,y1,row,col,array):
     return True
  return False

 def three_line(self,x1,y1,x2,y2,array):
  '''三条线相连,转弯两次'''
  for row1 in range(10):
   for col1 in range(14):
    for row2 in range(10):
     for col2 in range(14):
      if array[row1][col1] == array[row2][col2] == 0 and self.row_zero(x1,y1,row1,col1,array) and self.row_zero(x2,y2,row2,col2,array) and self.col_zero(row1,col1,row2,col2,array):
       return True
      if array[row1][col1] == array[row2][col2] == 0 and self.col_zero(x1,y1,row1,col1,array) and self.col_zero(x2,y2,row2,col2,array) and self.row_zero(row1,col1,row2,col2,array):
       return True
      if array[row1][col1] == array[row2][col2] == 0 and self.row_zero(x2,y2,row1,col1,array) and self.row_zero(x1,y1,row2,col2,array) and self.col_zero(row1,col1,row2,col2,array):
       return True
      if array[row1][col1] == array[row2][col2] == 0 and self.col_zero(x2,y2,row1,col1,array) and self.col_zero(x1,y1,row2,col2,array) and self.row_zero(row1,col1,row2,col2,array):
       return True
  return False


 def mouse_click(self,x,y):

  top = (x - 1) * 38 + (x - 2)
  left = (y - 1) * 38 + (y - 2)
  right = y * 38 + (y - 1)
  bottom = x * 38 + (x - 1)
  if top < 0:
   top = 0
  if left < 0:
   left = 0

  self.m.press(int(417+(left+right)/2) ,int(289+(top+bottom)/2) )

 def find_same_img(self,array):

  for x1 in range(1,9):
   for y1 in range(1,13):
    if array[x1][y1] == 0:
     continue
    for x2 in range(1,9):
     for y2 in range(1,13):
      if x1==x2 and y1 == y2:
       continue
      if array[x2][y2] == 0 :
       continue
      if array[x1][y1] != array[x2][y2] :
       continue
      if array[x1][y1] ==array[x2][y2] and (self.row_zero(x1,y1,x2,y2,array) or self.col_zero(x1,y1,x2,y2,array) or self.two_line(x1,y1,x2,y2,array) or self.three_line(x1,y1,x2,y2,array)):
       print("可消除!x{}y{} 和 x{}y{}".format(x1,y1,x2,y2))
       self.mouse_click(x1,y1)
       time.sleep(0.1)
       self.mouse_click(x2,y2)
       time.sleep(0.1)
       array[x1][y1]=array[x2][y2]=0



 def run(self):
  #找到游戏运行窗口
  self.find_game_wd(self.wdname)
  # 截图,切割成小图标
  self.get_img()
  print("切割完成")
  # 将图片矩阵转换成数字矩阵
  array = self.create_array()
  print(array)
  # 遍历矩阵,找到可消除项,点击消除
  for i in range(10):
   self.find_same_img(array)

  print(array)


if __name__ == '__main__':
 ga = GameAuxiliaries()
 ga.run()

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
用Python实现换行符转换的脚本的教程
Apr 16 Python
python获取外网ip地址的方法总结
Jul 02 Python
Python面向对象编程中关于类和方法的学习笔记
Jun 30 Python
python实现求解列表中元素的排列和组合问题
Mar 15 Python
python找出一个列表中相同元素的多个索引实例
Jun 11 Python
python opencv对图像进行旋转且不裁剪图片的实现方法
Jul 09 Python
python实现把两个二维array叠加成三维array示例
Nov 29 Python
pandas 中对特征进行硬编码和onehot编码的实现
Dec 20 Python
python GUI库图形界面开发之PyQt5窗口布局控件QStackedWidget详细使用方法
Feb 27 Python
Python socket连接中的粘包、精确传输问题实例分析
Mar 24 Python
Python数据相关系数矩阵和热力图轻松实现教程
Jun 16 Python
python 实现的截屏工具
May 08 Python
Django 路由控制的实现
Jul 17 #Python
详解python实现数据归一化处理的方式:(0,1)标准化
Jul 17 #Python
简单了解django索引的相关知识
Jul 17 #Python
python实现连连看辅助(图像识别)
Mar 25 #Python
Django中多种重定向方法使用详解
Jul 17 #Python
200行python代码实现2048游戏
Jul 17 #Python
Django后端接收嵌套Json数据及解析详解
Jul 17 #Python
You might like
php实现cookie加密的方法
2015/03/10 PHP
CodeIgniter针对数据库的连接、配置及使用方法
2016/03/03 PHP
PHP常用的三种设计模式汇总
2016/08/28 PHP
jquery 批量上传图片实现代码
2010/01/28 Javascript
15个款优秀的 jQuery 图片特效插件推荐
2011/11/21 Javascript
前台js调用后台方法示例
2013/12/02 Javascript
jquery左边浮动到一定位置时显示返回顶部按钮
2014/06/05 Javascript
如何正确使用javascript 来进行我们的程序开发
2014/06/23 Javascript
jQuery回调函数的定义及用法实例
2014/12/23 Javascript
多个jQuery版本共存的处理方案
2015/03/17 Javascript
js自定义select下拉框美化特效
2016/05/12 Javascript
javaScript如何跳出多重循环break、continue
2016/09/01 Javascript
vue实现裁切图片同时实现放大、缩小、旋转功能
2018/03/02 Javascript
nodejs多版本管理总结
2018/04/03 NodeJs
Node.js系列之安装配置与基本使用(1)
2019/08/30 Javascript
vue中改变滚动条样式的方法
2020/03/03 Javascript
Javascript作用域和作用域链原理解析
2020/03/03 Javascript
jQuery+Ajax+js实现请求json格式数据并渲染到html页面操作示例
2020/06/02 jQuery
vue2.0 解决抽取公用js的问题
2020/07/31 Javascript
在Python中使用模块的教程
2015/04/27 Python
python学习之编写查询ip程序
2016/02/27 Python
python中的selenium安装的步骤(浏览器自动化测试框架)
2020/03/17 Python
解决python和pycharm安装gmpy2 出现ERROR的问题
2020/08/28 Python
教你如何用python操作摄像头以及对视频流的处理
2020/10/12 Python
Python的Tqdm模块实现进度条配置
2021/02/24 Python
米兰网婚纱礼服法国网上商店:Milanoo法国
2016/08/20 全球购物
会计电算化个人自我评价
2013/11/17 职场文书
2014年三万活动总结
2014/04/26 职场文书
法人委托书范本
2014/09/15 职场文书
运动会广播稿50字-100字
2014/10/11 职场文书
解除租房协议书
2014/12/03 职场文书
小学生作文评语集锦
2014/12/25 职场文书
节约用电通知
2015/04/25 职场文书
在校学生证明格式
2015/06/24 职场文书
2015年工会工作总结范文
2015/07/23 职场文书
Java生成日期时间存入Mysql数据库的实现方法
2022/03/03 Java/Android