模板匹配是一種在圖像中尋找特定模式的技術(shù)。它通過(guò)滑動(dòng)一個(gè)模板圖像(較小的圖像)在輸入圖像上進(jìn)行比較,找到最相似的區(qū)域。本實(shí)驗(yàn)提供了一個(gè)簡(jiǎn)單的模板匹配案例,并將其封裝為一個(gè)自定義函數(shù) performTemplateMatching,方便快速移植和使用。
源代碼地址:https://gitee.com/LockzhinerAI/LockzhinerVisionModule/tree/master/Cpp_example/C02_TemplateMatching
1. 基本知識(shí)講解
1.1 模板匹配的重要性
目標(biāo)檢測(cè):模板匹配可以用于檢測(cè)圖像中的特定對(duì)象。
應(yīng)用場(chǎng)景:廣泛應(yīng)用于物體識(shí)別、工業(yè)自動(dòng)化、機(jī)器人導(dǎo)航等領(lǐng)域。
局限性:模板匹配對(duì)旋轉(zhuǎn)、縮放和光照變化較為敏感,因此通常需要結(jié)合其他技術(shù)來(lái)提高魯棒性。
1.2 模板匹配的流程
獲取輸入圖像和模板圖像。
使用模板匹配算法(如歸一化互相關(guān) NCC)計(jì)算相似度。
找到匹配結(jié)果中的最大值及其位置。
根據(jù)相似度閾值判斷匹配是否成功。
繪制矩形框標(biāo)記匹配區(qū)域并顯示結(jié)果。
2. API 文檔
2.1 頭文件
#include
2.2 在輸入圖像中搜索模板圖像的位置
voidmatchTemplate(InputArrayimage,InputArraytempl,OutputArrayresult,intmethod);
參數(shù):
image:待搜索的圖像。
templ:模板圖像。
result:搜索結(jié)果。
method:搜索方法,可以是 TM_SQDIFF、TM_SQDIFF_NORMED、TM_CCORR、TM_CCORR_NORMED、TM_CCOEFF、TM_CCOEFF_NORMED 之一。
返回值:
無(wú)
2.3 獲取匹配結(jié)果中的最大值及其位置
voidminMaxLoc(InputArraysrc,double*minVal,double*maxVal,Point*minLoc,Point*maxLoc,InputArraymask=noArray());
參數(shù):
src:輸入矩陣。
minVal:最小值。
maxVal:最大值。
minLoc:最小值位置。
maxLoc:最大值位置。
mask:可選的掩碼矩陣,用于指定要搜索的像素范圍。
返回值:
無(wú)
3. 綜合代碼解析
3.1 流程圖
3.2 代碼解釋
模板匹配函數(shù)應(yīng)用
doublesimilarityThreshold=0.7;// 相似度閾值
booluseGrayscale=false; // 是否使用灰度處理
boolmatchSuccess=performTemplateMatching(img,templ,similarityThreshold,img,useGrayscale);
模板匹配函數(shù)具體參數(shù)定義如下所示。
boolperformTemplateMatching(constMat&inputImage,constMat&templateImage,doublethreshold,Mat&outputImage,boolisGrayscale=false);
參數(shù):
inputImage:輸入圖像。
templateImage:模板圖像。
threshold:相似度閾值。
outputImage:輸出圖像。
isGrayscale:是否進(jìn)行灰度處理。
返回值:
true:匹配成功。
false:匹配失敗。
輸出結(jié)果
edit.Print(img);
3.3 代碼實(shí)現(xiàn)
#include
#include
#include
usingnamespacecv;
usingnamespacestd;
// 模板匹配函數(shù)
boolperformTemplateMatching(constMat&inputImage,constMat&templateImage,doublethreshold,Mat&outputImage,boolisGrayscale=false)
{
// 確保模板圖像比輸入圖像小
if(templateImage.rows>inputImage.rows||templateImage.cols>inputImage.cols)
{
cout<<"模板圖像不能大于輸入圖像!"<<endl;
returnfalse;
}
// 創(chuàng)建用于匹配的圖像副本
Mattempl=templateImage.clone();
Matimg=inputImage.clone();
// 如果選擇灰度處理,則將輸入圖像和模板圖像轉(zhuǎn)換為灰度
if(isGrayscale)
{
if(img.channels()==3)
{
cvtColor(img,img,COLOR_BGR2GRAY);
}
if(templ.channels()==3)
{
cvtColor(templ,templ,COLOR_BGR2GRAY);
}
}
// 打印調(diào)試信息
cout<<"輸入圖像尺寸: "<<img.size()<<", 通道數(shù): "<<img.channels()<<endl;
cout<<"模板圖像尺寸: "<<templ.size()<<", 通道數(shù): "<<templ.channels()<<endl;
// 創(chuàng)建結(jié)果矩陣,用于存儲(chǔ)匹配結(jié)果
intresultRows=img.rows-templ.rows+1;
intresultCols=img.cols-templ.cols+1;
if(resultRows<=0||resultCols<=0)
{
cout<<"結(jié)果矩陣尺寸無(wú)效!請(qǐng)檢查輸入圖像和模板圖像的尺寸。"<<endl;
returnfalse;
}
Matresult(resultRows,resultCols,CV_32FC1);
// 使用歸一化互相關(guān)(NCC)方法進(jìn)行模板匹配
doublestart=static_cast<double>(getTickCount());
matchTemplate(img,templ,result,TM_CCOEFF_NORMED);
doubleend=static_cast<double>(getTickCount());
doubleelapsedTime=(end-start)/getTickFrequency();
cout<<"matchTemplate 運(yùn)行時(shí)間: "<<elapsedTime<<" 秒"<<endl;
// 找到匹配結(jié)果中的最大值及其位置
doubleminVal,maxVal;
PointminLoc,maxLoc;
minMaxLoc(result,&minVal,&maxVal,&minLoc,&maxLoc);
// 輸出相似度閾值判斷
if(maxVal>=threshold)
{
cout<<"匹配成功!最大相似度: "<<maxVal<<endl;
// 繪制矩形框標(biāo)記匹配區(qū)域
rectangle(outputImage,maxLoc,Point(maxLoc.x+templ.cols,maxLoc.y+templ.rows),Scalar(0,255,0),2);
returntrue;
}
else
{
cout<<"匹配失??!最大相似度: "<<maxVal<<endl;
returnfalse;
}
}
intmain(intargc,char**argv)
{
// 檢查命令行參數(shù)數(shù)量是否正確
if(argc!=2)
{
cout<<"用法: "<<argv[0]<<" <模板圖像路徑>"<<endl;
return-1;
}
// 聲明并初始化變量
stringtemplateImagePath=argv[1];// 模板圖像路徑
lockzhiner_vision_module::Editedit;
if(!edit.StartAndAcceptConnection())
{
std::cerr<<"Error: Failed to start and accept connection."<<std::endl;
returnEXIT_FAILURE;
}
std::cout<<"Device connected successfully."<<std::endl;
// 初始化攝像頭
cv::VideoCapturecap;
intwidth=320; // 設(shè)置攝像頭分辨率寬度
intheight=240;// 設(shè)置攝像頭分辨率高度
cap.set(cv::CAP_PROP_FRAME_WIDTH,width);
cap.set(cv::CAP_PROP_FRAME_HEIGHT,height);
// 打開(kāi)攝像頭設(shè)備
cap.open(0);// 參數(shù) 0 表示默認(rèn)攝像頭設(shè)備
if(!cap.isOpened())
{
std::cerr<<"Error: Could not open camera."<<std::endl;
returnEXIT_FAILURE;
}
// 加載模板圖像
Mattempl=imread(templateImagePath,IMREAD_COLOR);// 默認(rèn)加載彩色模板
if(templ.empty())
{
std::cerr<<"Error: Could not load template image."<<std::endl;
returnEXIT_FAILURE;
}
while(true)
{
// 讀取輸入圖像
cv::Matimg;
cap>>img;
if(img.empty())
{
std::cerr<<"Error: Captured frame is empty."<<std::endl;
break;
}
// 調(diào)用模板匹配函數(shù)
doublesimilarityThreshold=0.7;// 相似度閾值
booluseGrayscale=false; // 是否使用灰度處理
boolmatchSuccess=performTemplateMatching(img,templ,similarityThreshold,img,useGrayscale);
// 顯示結(jié)果
imshow("Template Matching Result",img);
if(waitKey(1)==27)
{// 按 ESC 鍵退出
break;
}
// 輸出圖像
edit.Print(img);
}
return0;
}
4. 編譯過(guò)程
4.1 編譯環(huán)境搭建
請(qǐng)確保你已經(jīng)按照開(kāi)發(fā)環(huán)境搭建指南正確配置了開(kāi)發(fā)環(huán)境。
同時(shí)以正確連接開(kāi)發(fā)板。
4.2 Cmake介紹
# CMake最低版本要求
cmake_minimum_required(VERSION3.10)
project(test-TemplateMatching)
set(CMAKE_CXX_STANDARD17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
# 定義項(xiàng)目根目錄路徑
set(PROJECT_ROOT_PATH"${CMAKE_CURRENT_SOURCE_DIR}/../..")
message("PROJECT_ROOT_PATH = "${PROJECT_ROOT_PATH})
include("${PROJECT_ROOT_PATH}/toolchains/arm-rockchip830-linux-uclibcgnueabihf.toolchain.cmake")
# 定義 OpenCV SDK 路徑
set(OpenCV_ROOT_PATH"${PROJECT_ROOT_PATH}/third_party/opencv-mobile-4.10.0-lockzhiner-vision-module")
set(OpenCV_DIR"${OpenCV_ROOT_PATH}/lib/cmake/opencv4")
find_package(OpenCV REQUIRED)
set(OPENCV_LIBRARIES"${OpenCV_LIBS}")
# 定義 LockzhinerVisionModule SDK 路徑
set(LockzhinerVisionModule_ROOT_PATH"${PROJECT_ROOT_PATH}/third_party/lockzhiner_vision_module_sdk")
set(LockzhinerVisionModule_DIR"${LockzhinerVisionModule_ROOT_PATH}/lib/cmake/lockzhiner_vision_module")
find_package(LockzhinerVisionModule REQUIRED)
# 基本圖像處理示例
add_executable(Test-TemplateMatching TemplateMatching.cc)
target_include_directories(Test-TemplateMatching PRIVATE${LOCKZHINER_VISION_MODULE_INCLUDE_DIRS})
target_link_libraries(Test-TemplateMatching PRIVATE${OPENCV_LIBRARIES}${LOCKZHINER_VISION_MODULE_LIBRARIES})
install(
TARGETS Test-TemplateMatching
RUNTIME DESTINATION .
)
4.3 編譯項(xiàng)目
使用 Docker Destop 打開(kāi) LockzhinerVisionModule 容器并執(zhí)行以下命令來(lái)編譯項(xiàng)目
# 進(jìn)入Demo所在目錄
cd/LockzhinerVisionModuleWorkSpace/LockzhinerVisionModule/Cpp_example/A01_capture
# 創(chuàng)建編譯目錄
rm-rfbuild &&mkdirbuild &&cdbuild
# 配置交叉編譯工具鏈
exportTOOLCHAIN_ROOT_PATH="/LockzhinerVisionModuleWorkSpace/arm-rockchip830-linux-uclibcgnueabihf"
# 使用cmake配置項(xiàng)目
cmake ..
# 執(zhí)行編譯項(xiàng)目
make-j8&&makeinstall
在執(zhí)行完上述命令后,會(huì)在build目錄下生成可執(zhí)行文件。
5. 例程運(yùn)行示例
5.1 準(zhǔn)備工作
下載凌智視覺(jué)模塊圖片傳輸助手:點(diǎn)擊下載
我們首先需要進(jìn)行圖像采集,采集一個(gè)480*320分辨率下的模板照片。
5.2 運(yùn)行過(guò)程
在凌智視覺(jué)模塊中輸入以下命令:
chmod777Test-TemplateMatching template_0.png
./Test-TemplateMatching
5.3 運(yùn)行結(jié)果
運(yùn)行結(jié)果如下:
模板照片為:
6. 總結(jié)
通過(guò)上述內(nèi)容,我們?cè)敿?xì)介紹了模板匹配的流程及相關(guān) API 的使用方法,包括:
圖像讀取:加載輸入圖像和模板圖像。
模板匹配:使用歸一化互相關(guān)方法計(jì)算相似度。
查找極值:獲取最佳匹配位置。
繪制與顯示:標(biāo)記匹配區(qū)域并顯示結(jié)果。希望這份文檔能幫助您更好地理解和實(shí)現(xiàn)模板匹配功能!注意事項(xiàng):
在本次例程中攝像頭需要運(yùn)行在480*320分辨率下,否則會(huì)出現(xiàn)幀率過(guò)低的情況。如需更高幀率,可酌情再降低分辨率。
-
圖像
+關(guān)注
關(guān)注
2文章
1092瀏覽量
41010 -
視覺(jué)識(shí)別
+關(guān)注
關(guān)注
3文章
101瀏覽量
16970
發(fā)布評(píng)論請(qǐng)先 登錄
基于LockAI視覺(jué)識(shí)別模塊:C++圖像采集例程
基于LockAI視覺(jué)識(shí)別模塊:C++模板匹配
基于LockAI視覺(jué)識(shí)別模塊:C++多模板匹配
基于LockAI視覺(jué)識(shí)別模塊:C++二維碼識(shí)別
基于LockAI視覺(jué)識(shí)別模塊:C++條碼識(shí)別
labview 視覺(jué) 多模板匹配 教程
labview 視覺(jué) 多模板匹配 教程
基于LockAI視覺(jué)識(shí)別模塊:C++圖像采集例程

基于LockAI視覺(jué)識(shí)別模塊:C++圖像的基本運(yùn)算

基于LockAI視覺(jué)識(shí)別模塊:C++使用圖像的統(tǒng)計(jì)信息

# 基于LockAI視覺(jué)識(shí)別模塊:C++尋找色塊

基于LockAI視覺(jué)識(shí)別模塊:C++多模板匹配

基于LockAI視覺(jué)識(shí)別模塊:C++二維碼識(shí)別

基于LockAI視覺(jué)識(shí)別模塊:C++條碼識(shí)別

評(píng)論