如何将图片中的数据插入二维数组？

Question

我正在制作一个可以玩游戏的 AI 连接 4，来自游戏的一种状态的图片，例如：click to see

下面的这个脚本，正在检测图片中的红色元素：

import cv2
import numpy as np
 
img = cv2.imread('connect.png')
 
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
 
#get red color 
lower_range = np.array([169, 100, 100])
upper_range = np.array([189, 255, 255])
 
mask = cv2.inRange(hsv, lower_range, upper_range)
 
cv2.imshow('image', img)
cv2.imshow('mask', mask)
 
cv2.waitKey(0)
cv2.destroyAllWindows()

我想将这些数据插入到一个二维数组中，以便能够将该数组用作游戏状态并确定 AI 应该采取的行动。

我试图在 Stack Overflow 和 Internet 上找到解决方案，但总的来说我没有找到任何相关信息。

Answer 1

这是一种读取图片并将其转换为 二维 numpy 数组的方法，np.array(in_image):

import numpy as np
import skimage
from skimage import io, transform


path = "C:/my/path/"
pic = 'myPic.png'
imgName = path+pic

in_image_0 = skimage.io.imread(imgName)                   # read the image
in_image_1 = skimage.color.rgb2gray(in_image_0)           # transform it to grey-scale
in_image_2 = skimage.transform.rescale(in_image_1, 0.5)   # change the resolution
in_image_3 = np.flipud(np.array(in_image_2))              # make a numpy array and flip it up/down