在傳統(tǒng)計(jì)算機(jī)視覺場(chǎng)景中，顏色識(shí)別是目標(biāo)檢測(cè)和分割的重要手段之一。通過識(shí)別特定顏色的色塊，可以在相對(duì)純凈的背景下快速定位目標(biāo)區(qū)域。本實(shí)驗(yàn)提供了一個(gè)簡(jiǎn)單的色塊識(shí)別案例，并將其封裝為一個(gè)自定義函數(shù) find_blobs，方便快速移植和使用。

源代碼地址：https://gitee.com/LockzhinerAI/LockzhinerVisionModule/tree/master/Cpp_example/C01_find_blobs

1. 基本知識(shí)講解

1.1 色塊識(shí)別的重要性

顏色特征提?。侯伾且环N重要的視覺特征，尤其在背景較為單一的情況下，能夠快速區(qū)分目標(biāo)區(qū)域。

應(yīng)用場(chǎng)景：廣泛應(yīng)用于機(jī)器人導(dǎo)航、工業(yè)自動(dòng)化、物體跟蹤等領(lǐng)域。

HSV 顏色空間：相比于 RGB 顏色空間，HSV 更適合用于顏色識(shí)別，因?yàn)樗梢詫㈩伾畔ⅲ℉ue）、飽和度（Saturation）和亮度（Value）分離，便于設(shè)置閾值。

1.2 色塊識(shí)別的流程

獲取圖像。

將圖像從 BGR 轉(zhuǎn)換為 HSV 顏色空間。

創(chuàng)建二值掩碼，篩選出符合顏色范圍的像素。

使用形態(tài)學(xué)操作清除噪聲。

查找輪廓并篩選符合條件的色塊。

計(jì)算外接矩形和中心點(diǎn)。

繪制結(jié)果并輸出。

2. API文檔

2.1 頭文件

#include

2.2 生成掩碼

cv::inRange(src,lowerb,upperb,dst);

參數(shù)說明：

src：輸入圖像，可以是單通道或三通道的圖像。

lowerb：顏色下界，是一個(gè)Scalar對(duì)象，表示要查找的顏色的下限。

upperb：顏色上界，是一個(gè)Scalar對(duì)象，表示要查找的顏色的上限。

dst：輸出圖像，是一個(gè)單通道的8位無符號(hào)整數(shù)圖像，表示生成的掩碼。

返回值：

無

2.3 創(chuàng)建形態(tài)學(xué)操作所需的結(jié)構(gòu)元素核

cv::getStructuringElement(shape,ksize,anchor);

參數(shù)說明：

shape：核形狀，可以是RECT、CROSS、ELLIPSE等。

ksize：核大小，是一個(gè)Size對(duì)象，表示核的寬度和高度。

anchor：錨點(diǎn)，是一個(gè)Point對(duì)象，表示核的錨點(diǎn)位置。

返回值：

返回一個(gè)核，是一個(gè)Mat對(duì)象。

2.4 形態(tài)學(xué)操作：清除噪聲

cv::morphologyEx(src,dst,op,kernel,anchor,iterations,borderType,borderValue);

參數(shù)說明：

src：輸入圖像，可以是單通道或三通道的圖像。

dst：輸出圖像，是一個(gè)單通道的8位無符號(hào)整數(shù)圖像，表示生成的掩碼。

op：操作類型，可以是OPEN、CLOSE、GRADIENT、TOPHAT、BLACKHAT等。

kernel：核，是一個(gè)Mat對(duì)象，表示形態(tài)學(xué)操作的核。

anchor：錨點(diǎn)，是一個(gè)Point對(duì)象，表示核的錨點(diǎn)位置。

iterations：迭代次數(shù)，是一個(gè)整數(shù)，表示形態(tài)學(xué)操作的迭代次數(shù)。

borderType：邊界類型，可以是BORDER_CONSTANT、BORDER_REPLICATE、BORDER_REFLECT、BORDER_WRAP、BORDER_REFLECT_101等。

borderValue：邊界值，是一個(gè)Scalar對(duì)象，表示邊界區(qū)域的值。

返回值：

無

2.5 查找輪廓

cv::findContours(image,contours,hierarchy,mode,method,offset);

參數(shù)說明：

image：輸入圖像，可以是單通道或三通道的圖像。

contours：輸出參數(shù)，是一個(gè)vector>對(duì)象，表示輪廓的集合。

hierarchy：輸出參數(shù)，是一個(gè)vector對(duì)象，表示輪廓的層級(jí)關(guān)系。

mode：輪廓發(fā)現(xiàn)模式，可以是RETR_EXTERNAL、RETR_LIST、RETR_CCOMP、RETR_TREE等。

method：輪廓 approximation 方法，可以是CHAIN_APPROX_NONE、CHAIN_APPROX_SIMPLE、CHAIN_APPROX_TC89_L1、CHAIN_APPROX_TC89_KCOS等。

offset：輪廓偏移量，是一個(gè)Point對(duì)象，表示輪廓的偏移量。

返回值：

返回一個(gè)整數(shù)，表示輪廓的數(shù)量。

2.6 獲取輪廓的外接矩形

cv::boundingRect(points);

參數(shù)說明：

points：輸入?yún)?shù)，是一個(gè)vector對(duì)象，表示輪廓的點(diǎn)集合。

返回值：

返回一個(gè)Rect對(duì)象，表示輪廓的外接矩形。

2.7 計(jì)算矩陣矩

cv::moments(array,binaryImage);

參數(shù)說明：

array：輸入?yún)?shù)，是一個(gè)Mat對(duì)象，表示輸入的矩陣。

binaryImage：輸入?yún)?shù)，是一個(gè)布爾值，表示是否將輸入的矩陣轉(zhuǎn)換為二值矩陣。

返回值：

返回一個(gè) Moments對(duì)象，表示矩陣的矩。

2.8 繪制矩形框

cv::rectangle(img,pt1,pt2,color,thickness,lineType,shift);

參數(shù)說明：

img：輸入?yún)?shù)，是一個(gè)Mat對(duì)象，表示輸入的圖像。

pt1：輸入?yún)?shù)，是一個(gè)Point對(duì)象，表示矩形的左上角點(diǎn)。

pt2：輸入?yún)?shù)，是一個(gè)Point對(duì)象，表示矩形的右下角點(diǎn)。

color：輸入?yún)?shù)，是一個(gè)Scalar對(duì)象，表示矩形的顏色。

thickness：輸入?yún)?shù)，是一個(gè)整數(shù)，表示矩形的線寬。

lineType：輸入?yún)?shù)，是一個(gè)整數(shù)，表示矩形的線類型。

shift：輸入?yún)?shù)，是一個(gè)整數(shù)，表示坐標(biāo)的精度。

返回值：

無

2.9 繪制圓

cv::circle(img,center,radius,color,thickness,lineType,shift);

參數(shù)說明：

img：輸入?yún)?shù)，是一個(gè)Mat對(duì)象，表示輸入的圖像。

center：輸入?yún)?shù)，是一個(gè)Point對(duì)象，表示圓心。

radius：輸入?yún)?shù)，是一個(gè)整數(shù)，表示圓的半徑。

color：輸入?yún)?shù)，是一個(gè)Scalar對(duì)象，表示圓的顏色。

thickness：輸入?yún)?shù)，是一個(gè)整數(shù)，表示圓的線寬。

lineType：輸入?yún)?shù)，是一個(gè)整數(shù)，表示圓的線類型。

shift：輸入?yún)?shù)，是一個(gè)整數(shù)，表示坐標(biāo)的精度。

返回值：

無

3. 綜合代碼介紹

3.1 流程圖

3.2 核心代碼解析

BGR轉(zhuǎn)HSV

cv::cvtColor(image,hsv_image,cv::COLOR_BGR2HSV);

閾值分割

cv::inRange(hsv_image,lower_bound,upper_bound,mask););

形態(tài)學(xué)處理

cv::Matkernel=cv::getStructuringElement(cv::MORPH_RECT,cv::Size(kernel_size,kernel_size));
cv::morphologyEx(mask,mask,cv::MORPH_OPEN,kernel);

查找輪廓

cv::findContours(mask,contours,cv::RETR_EXTERNAL,cv::CHAIN_APPROX_SIMPLE);

自定義函數(shù)參數(shù)如下所示

std::vector<std::vector<cv::Point>>find_blobs(
constcv::Mat&image,
constcv::Scalar&lower_bound,
constcv::Scalar&upper_bound,
intmin_area=100,
intkernel_size=5);

參數(shù)說明：

image：輸入?yún)?shù)，是一個(gè)Mat對(duì)象，表示輸入的圖像。

lower_bound：輸入?yún)?shù)，是一個(gè)Scalar對(duì)象，表示顏色下界。

upper_bound：輸入?yún)?shù)，是一個(gè)Scalar對(duì)象，表示顏色上界。

min_area：輸入?yún)?shù)，是一個(gè)整數(shù)，表示最小面積。

kernel_size：輸入?yún)?shù)，是一個(gè)整數(shù)，表示核大小。

返回值：

返回一個(gè)vector>對(duì)象，表示找到的色塊的點(diǎn)集合。

3.3 完整代碼實(shí)現(xiàn)

#include
#include
#include
#include

std::vector<std::vector<cv::Point>>find_blobs(
constcv::Mat&image,
constcv::Scalar&lower_bound,
constcv::Scalar&upper_bound,
intmin_area=100,
intkernel_size=5)
{

// 轉(zhuǎn)換為 HSV 顏色空間
cv::Mathsv_image;
cv::cvtColor(image,hsv_image,cv::COLOR_BGR2HSV);

// 創(chuàng)建二值掩碼
cv::Matmask;
cv::inRange(hsv_image,lower_bound,upper_bound,mask);

// 形態(tài)學(xué)操作：清除噪聲
cv::Matkernel=cv::getStructuringElement(cv::MORPH_RECT,cv::Size(kernel_size,kernel_size));
cv::morphologyEx(mask,mask,cv::MORPH_OPEN,kernel);

// 查找輪廓
std::vector<std::vector<cv::Point>>contours;
cv::findContours(mask,contours,cv::RETR_EXTERNAL,cv::CHAIN_APPROX_SIMPLE);

// 篩選符合條件的色塊
std::vector<std::vector<cv::Point>>filtered_contours;
for(constauto&contour:contours)
{
cv::Rectbounding_rect=cv::boundingRect(contour);
if(bounding_rect.area()>=min_area)
{
filtered_contours.push_back(contour);
}
}
returnfiltered_contours;
}

intmain()
{
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=640; // 設(shè)置攝像頭分辨率寬度
intheight=480;// 設(shè)置攝像頭分辨率高度
cap.set(cv::CAP_PROP_FRAME_WIDTH,width);
cap.set(cv::CAP_PROP_FRAME_HEIGHT,height);

// 打開攝像頭設(shè)備
cap.open(0);// 參數(shù) 0 表示默認(rèn)攝像頭設(shè)備
if(!cap.isOpened())
{
std::cerr<<"Error: Could not open camera."<<std::endl;
returnEXIT_FAILURE;
}
while(true)
{
cv::Matimage;// 存儲(chǔ)每一幀圖像
cap>>image; // 獲取新的一幀

// 定義顏色閾值（例如紅色）
cv::Scalarlower_red(170,100,100);// 紅色下界
cv::Scalarupper_red(179,255,255);// 紅色上界

// 調(diào)用 find_blobs 函數(shù)
intmin_area=100; // 最小面積閾值
intkernel_size=1;// 形態(tài)學(xué)操作核大小
std::vector<std::vector<cv::Point>>blobs=find_blobs(image,lower_red,upper_red,min_area,kernel_size);

// 繪制和打印檢測(cè)到的色塊
for(constauto&contour:blobs)
{
// 計(jì)算外接矩形框
cv::Rectbounding_rect=cv::boundingRect(contour);

// 繪制矩形框
cv::rectangle(image,bounding_rect,cv::Scalar(0,255,0),2);

// 計(jì)算中心點(diǎn)
cv::Momentsmoments=cv::moments(contour);
intcx=moments.m10/moments.m00;
intcy=moments.m01/moments.m00;

// 繪制中心點(diǎn)
cv::circle(image,cv::Point(cx,cy),5,cv::Scalar(0,0,255),-1);

// 打印信息
std::cout<<"Blob detected at ("<<cx<<", "<<cy<<") with area "<<bounding_rect.area()<<std::endl;
}
edit.Print(image);
}
return0;
}

4. 編譯過程

4.1 編譯環(huán)境搭建

請(qǐng)確保你已經(jīng)按照開發(fā)環(huán)境搭建指南正確配置了開發(fā)環(huán)境。

同時(shí)以正確連接開發(fā)板。

4.2 Cmake介紹

# CMake最低版本要求
cmake_minimum_required(VERSION3.10)

project(test-find-blobs)

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-find-blobs find_blobs.cc)
target_include_directories(Test-find-blobs PRIVATE${LOCKZHINER_VISION_MODULE_INCLUDE_DIRS})
target_link_libraries(Test-find-blobs PRIVATE${OPENCV_LIBRARIES}${LOCKZHINER_VISION_MODULE_LIBRARIES})

install(
TARGETS Test-find-blobs
RUNTIME DESTINATION .
)

4.3 編譯項(xiàng)目

使用 Docker Destop 打開 LockzhinerVisionModule 容器并執(zhí)行以下命令來編譯項(xiàng)目

# 進(jìn)入Demo所在目錄
cd/LockzhinerVisionModuleWorkSpace/LockzhinerVisionModule/Cpp_example/C01_find_blobs
# 創(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 運(yùn)行過程

chmod777Test-find-blobs
./Test-find-blobs

5.2 運(yùn)行效果

6. 總結(jié)

通過上述內(nèi)容，我們?cè)敿?xì)介紹了色塊識(shí)別的流程及相關(guān) API 的使用方法，包括：

生成掩碼：篩選符合顏色范圍的像素。

形態(tài)學(xué)操作：清除噪聲。

查找輪廓：獲取目標(biāo)區(qū)域的輪廓。

篩選與繪制：篩選符合條件的色塊并繪制外接矩形和中心點(diǎn)。