差值平方和匹配_机器学习实战 | 简单目标识别与意图分析之模板匹配
(點擊上方快速關(guān)注并設(shè)置為星標,一起學(xué)Python)
一天,我正在學(xué)校樓下擼貓,同學(xué)發(fā)來消息,他的老師給了他一個研究課題,大致的方向是對圖片或者視頻里面的內(nèi)容進行識別,然后判斷意圖,而且舉了個例子,兩方交鋒,現(xiàn)在根據(jù)無人機拍攝或者衛(wèi)星圖像,針對對方坦克集團的陣型來分析對方坦克的作戰(zhàn)意圖。但他對課題實現(xiàn)一臉懵逼,所以過來求救,我一聽,真的好高大上啊!為發(fā)揚同學(xué)間團結(jié)有愛、互幫互助的精神,我就幫忙研究研究吧(每天一瓶飲料,做錯了別怪我。。。。)
這樣講文章思路不清晰,我先畫個流程圖來描述一下我的工作,避免大家繞來繞去的看的發(fā)懵:
初步目標識別
同學(xué)說老師要他先實現(xiàn)這個目標識別,我腦子里第一時間想到了Python3里面處理圖片的超強第三方庫OpenCV,里面的模板匹配有好多算法可以用呢,比如平方差匹配、相關(guān)性匹配、相關(guān)性系數(shù)匹配還有匹配效果最好的標準相關(guān)匹配等等,然后模型的話最近正好在研究Unity寫游戲,那就拿里面的坦克模型來測試,先來張截圖測試下,一言不合上代碼:
tpl =cv.imread(r"D://PythonPicTemplate/pythinPIC/tankTemplate2.jpg")target = cv.imread("D://PythonPicTemplate/pythinPIC/tanks2.jpg")
cv.namedWindow('template image', cv.WINDOW_NORMAL)
cv.imshow("template image", tpl)
cv.namedWindow('target image', cv.WINDOW_NORMAL)
cv.imshow("target image", target)
#methods = [cv.TM_CCOEFF, cv.TM_CCOEFF_NORMED, cv.TM_CCORR,cv.TM_CCORR_NORMED, cv.TM_SQDIFF, cv.TM_SQDIFF_NORMED]
methods =[cv.TM_CCORR_NORMED ,cv.TM_CCORR_NORMED,cv.TM_CCORR_NORMED,cv.TM_CCORR_NORMED, cv.TM_CCORR_NORMED]
'''''''
差值平方和匹配 CV_TM_SQDIFF
標準化差值平方和匹配 CV_TM_SQDIFF_NORMED
相關(guān)匹配 CV_TM_CCORR
標準相關(guān)匹配 CV_TM_CCORR_NORMED
相關(guān)匹配 CV_TM_CCOEFF
標準相關(guān)匹配 CV_TM_CCOEFF_NORMED
'''th, tw = tpl.shape[:2]
i =0
tl = []
br = []
for md in methods:
#print(md)
result = cv.matchTemplate(target, tpl, md)
min_val, max_val, min_loc, max_loc = cv.minMaxLoc(result)
if md == cv.TM_SQDIFF_NORMED:
tl.append(min_loc)
else:
tl.append(max_loc)
br.append((tl[i][0]+tw, tl[i][1]+th))
cv.rectangle(target, tl[i], br[i], (0, 0, 255), 2)
i += 1
cv.namedWindow("match-" + np.str(md), cv.WINDOW_NORMAL)
cv.imshow("match-" + np.str(md), target)
模板匹配的大致原理
模板匹配實現(xiàn)的思想也是很簡單很暴力的,就是拿著模板圖片在原圖中從左上至右下依次滑動,直到遇到某個區(qū)域的相似度低于我們設(shè)定的閾值,那么我們就認為該區(qū)域與模板匹配了,并把它標記出來。但實驗證明,模板配在原圖摳出模板圖的形式下準確率才比較高,不然的話可能準確度就不太高了。
綜上所述,模板匹配的優(yōu)點在于在模板的基礎(chǔ)上能匹配的方便快速,但缺點實在是太多了,例如,
1.模板的大小會影響匹配的精度;
2.干擾現(xiàn)象嚴重,坦克藏在樹林里很影響識別效果;
3.在實際運用中,模板的方向都必須要固定才能匹配成功;
4.以上都還好說,最重要的是我都知道模板了,那我還寫代碼在圖片里識別個毛啊,對于這個課題直接圈出來不就行了。
所以我還是太天真了,模板匹配的優(yōu)點要用,但目標識別的方法必須要改進一下,不然看著都傻,同學(xué)老師估計也會爆發(fā)。下一節(jié)我們繼續(xù)研究具體的改進方法。
代碼已上傳至GitHub,歡迎star,歡迎討論:
GitHub:https://github.com/wangwei39120157028/Machine_Learning_research_on_simple_target_recognition_and_intention_analysis
(完)
看完本文有收獲?請轉(zhuǎn)發(fā)分享給更多人
關(guān)注「Python那些事」,做全棧開發(fā)工程師
點「在看」的人都變好看了哦總結(jié)
以上是生活随笔為你收集整理的差值平方和匹配_机器学习实战 | 简单目标识别与意图分析之模板匹配的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 酷派COOL 20s终于抛弃720P屏幕
- 下一篇: 12306加密传输_三大运营商发5G消息