自学内容网 自学内容网

MATLAB 识别色块和数量


前言

提示:这里可以添加本文要记录的大概内容:

项目需要:


在 MATLAB 中进行色块识别通常涉及图像处理技术,包括图像读取、预处理、颜色空间转换、阈值分割和形态学操作等步骤。以下是一个基本的示例代码,展示了如何识别图像中的特定色块。

假设你有一张包含多种颜色的图像,并且你想识别出特定颜色的色块(例如红色)。

步骤 1: 读取图像

% 读取图像
img = imread('your_image.jpg');

步骤 2: 转换为 HSV 颜色空间

HSV 颜色空间更适合颜色分割,因为它将颜色信息(色调、饱和度、亮度)分开。

% 将图像从 RGB 转换为 HSV 颜色空间
hsvImg = rgb2hsv(img);

步骤 3: 定义颜色范围

在 HSV 空间中定义你要识别的颜色的范围。例如,红色可能定义为:

% 定义红色的 HSV 范围
redHueRange = [0, 0.1];  % 色调范围
redSaturationRange = [0.5, 1];  % 饱和度范围
redValueRange = [0.5, 1];  % 亮度范围

步骤 4: 创建颜色掩码

使用逻辑掩码来标识图像中的红色区域。

% 创建颜色掩码
hueMask = (hsvImg(:,:,1) >= redHueRange(1)) & (hsvImg(:,:,1) <= redHueRange(2));
saturationMask = (hsvImg(:,:,2) >= redSaturationRange(1)) & (hsvImg(:,:,2) <= redSaturationRange(2));
valueMask = (hsvImg(:,:,3) >= redValueRange(1)) & (hsvImg(:,:,3) <= redValueRange(2));

colorMask = hueMask & saturationMask & valueMask;

步骤 5: 应用形态学操作(可选)

为了去除噪声和小的孤立区域,可以使用形态学操作,如膨胀和腐蚀。

% 形态学操作
se = strel('disk', 5);  % 创建一个结构元素
cleanedMask = imopen(colorMask, se);  % 开运算去除小区域
cleanedMask = imclose(cleanedMask, se);  % 闭运算填充小孔

步骤 6: 标记和显示结果

使用 bwlabel 标记连通区域,并显示结果。

% 标记连通区域
[labeledImage, numRegions] = bwlabel(cleanedMask);

% 显示结果
figure;
subplot(1, 3, 1);
imshow(img);
title('原始图像');

subplot(1, 3, 2);
imshow(colorMask);
title('颜色掩码');

subplot(1, 3, 3);
imshow(label2rgb(labeledImage, 'jet', 'k', 'shuffle'));
title(['识别到的色块数量: ', num2str(numRegions)]);

完整代码

% 读取图像
img = imread('your_image.jpg');

% 将图像从 RGB 转换为 HSV 颜色空间
hsvImg = rgb2hsv(img);

% 定义红色的 HSV 范围
redHueRange = [0, 0.1];  % 色调范围
redSaturationRange = [0.5, 1];  % 饱和度范围
redValueRange = [0.5, 1];  % 亮度范围

% 创建颜色掩码
hueMask = (hsvImg(:,:,1) >= redHueRange(1)) & (hsvImg(:,:,1) <= redHueRange(2));
saturationMask = (hsvImg(:,:,2) >= redSaturationRange(1)) & (hsvImg(:,:,2) <= redSaturationRange(2));
valueMask = (hsvImg(:,:,3) >= redValueRange(1)) & (hsvImg(:,:,3) <= redValueRange(2));

colorMask = hueMask & saturationMask & valueMask;

% 形态学操作
se = strel('disk', 5);  % 创建一个结构元素
cleanedMask = imopen(colorMask, se);  % 开运算去除小区域
cleanedMask = imclose(cleanedMask, se);  % 闭运算填充小孔

% 标记连通区域
[labeledImage, numRegions] = bwlabel(cleanedMask);

% 显示结果
figure;
subplot(1, 3, 1);
imshow(img);
title('原始图像');

subplot(1, 3, 2);
imshow(colorMask);
title('颜色掩码');

subplot(1, 3, 3);
imshow(label2rgb(labeledImage, 'jet', 'k', 'shuffle'));
title(['识别到的色块数量: ', num2str(numRegions)]);

这段代码提供了一个基本的框架,你可以根据具体需求调整颜色范围、形态学操作和结构元素的大小等参数。

在这里插入图片描述

步骤七 返回色块坐标

% 使用 regionprops 计算每个连通区域的属性
stats = regionprops(labeledImage, 'BoundingBox');
 
% 提取边界框坐标
boundingBoxes = [stats.BoundingBox];
 
% 显示结果,包括边界框
figure;
imshow(img);
hold on;
for k = 1:size(boundingBoxes, 1)
    rectangle('Position', boundingBoxes(k, :), 'EdgeColor', 'r', 'LineWidth', 2);
    % 如果需要,可以在这里打印或存储坐标
    fprintf('色块 %d 的坐标: [%f, %f, %f, %f]\n', k, boundingBoxes(k, 1), boundingBoxes(k, 2), boundingBoxes(k, 3), boundingBoxes(k, 4));
end
hold off;
title('识别到的色块及其边界框');

在这里插入图片描述
色块 1 的坐标: [160.500000, 156.500000, 44.000000, 43.000000]

总结

学习使人快乐!
音乐使人愉悦!
日积月累使人充实和自信!


原文地址:https://blog.csdn.net/ganhui13000/article/details/144410821

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!