快捷導(dǎo)航

python微信跳一跳系列之棋子定位顏色識(shí)別

更新時(shí)間：2018年02月26日 13:57:09 作者：艾克思工作室

這篇文章主要為大家詳細(xì)介紹了python微信跳一跳系列之棋子定位之顏色識(shí)別，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

python微信跳一跳，前言

這是python玩跳一跳系列博文中一篇，主要內(nèi)容是用顏色識(shí)別的方法來(lái)進(jìn)行跳跳小人的定位。

顏色識(shí)別

通過(guò)觀察，我們可以發(fā)現(xiàn)，盡管背景和棋子在不停的變化，但跳跳小人的形狀和顏色基本保持不變，對(duì)于形狀，我們?cè)谏弦黄┪闹幸呀?jīng)采用模板匹配的方法來(lái)進(jìn)行識(shí)別定位，效果非常好。這一篇博文就來(lái)對(duì)顏色識(shí)別進(jìn)行驗(yàn)證。

基本思路

用HSV顏色空間對(duì)輸入的圖片進(jìn)行處理，用某種指定的顏色進(jìn)行蒙版mask處理進(jìn)而得到二值化的黑白圖像，膨脹和腐蝕后去除噪點(diǎn)，對(duì)輪廓區(qū)域進(jìn)行計(jì)算，畫(huà)出圓心和質(zhì)心位置，并實(shí)現(xiàn)動(dòng)態(tài)的跟蹤。
其基本的步驟如下：
設(shè)定需要的顏色閾值
讀入圖像
轉(zhuǎn)化為HSV圖像
采用顏色的蒙版進(jìn)行二值化處理得到黑白圖像
降噪和輪廓處理
繪出圓心

python3.6代碼

import cv2 
import numpy as np 
import time

lower_blue = np.array([115,75,75]) #設(shè)定藍(lán)色的閾值
upper_blue = np.array([130,255,125])

frame=cv2.imread('001.png')

hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) #轉(zhuǎn)到HSV空間
mask_blue = cv2.inRange(hsv, lower_blue, upper_blue)
cnts = cv2.findContours(mask_blue, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)[-2] 
if len(cnts) > 0: 
 c = max(cnts, key = cv2.contourArea) #找到面積最大的輪廓
 ((x, y), radius) = cv2.minEnclosingCircle(c) #確定面積最大的輪廓的外接圓

 center= (int(x),int(y))
 cv2.circle(frame, center, int(radius+10), (0, 0, 255), 3) #畫(huà)出圓心 
 cv2.circle(frame, center, 3, (0, 0, 255), -1)
 cv2.circle(hsv, center, int(radius+10), (255, 255, 255), 3) #畫(huà)出圓心 
 cv2.circle(hsv, center, 3, (0, 0, 255), -1) 
 cv2.circle(mask_blue, center, int(radius+10), (255, 255, 255), 3) #畫(huà)出圓心 
 cv2.circle(mask_blue, center, 3, (0, 0, 255), -1) 

cv2.imshow('frame',frame)
cv2.imshow('hsv',hsv)
cv2.imshow('mask',mask_blue)
if cv2.waitKey(0)==ord('q'):
 cv2.destroyAllWindows()

靜態(tài)圖片識(shí)別效果